5175 2023. 04. 20 17:32:49 SleepyOverlord Szivárványszámok cpp17 Elfogadva 45/45 3ms 4668 KiB
#include <vector>
#include <string> 
#include <set> 
#include <map> 
#include <unordered_set>
#include <unordered_map>
#include <queue> 
#include <bitset> 
#include <stack>
#include <list>

#include <numeric> 
#include <algorithm> 
#include <random>
#include <chrono>

#include <cstdio>
#include <fstream>
#include <iostream> 
#include <sstream> 
#include <iomanip>
#include <climits>

#include <cctype>
#include <cmath> 
#include <ctime>
#include <cassert>

using namespace std;

#define ULL unsigned long long
#define LL long long
#define PII pair <int, int>
#define VB vector <bool>
#define VI vector <int>
#define VLL vector <LL>
#define VD vector <double>
#define VS vector <string>
#define VPII vector <pair <int, int> >
#define VVI vector < VI >
#define VVB vector < VB >
#define SI set < int >
#define USI unordered_set <int>
#define MII map <int, int>
#define UMII unordered_map <int, int>
#define MS multiset
#define US unordered_set
#define UM unordered_map
#define UMS unordered_multiset
#define UMM unordered_multimap

#define FORN(i, n) for(int i = 0; i < (n); ++i)
#define FOR(i, a, b) for(int i = (a); i <= (b); ++i)
#define FORD(i, a, b) for(int i = (a); i >= (b); --i)

#define SZ size()
#define BG begin() 
#define EN end() 
#define CL clear()
#define X first
#define Y second
#define RS resize
#define PB push_back
#define MP make_pair
#define ALL(x) x.begin(), x.end()
#define INS insert
#define ER erase
#define CNT count

//dp[i][j][0/1] - hany szivarvanyszam van, melynek i. szamjegye j es 
//monoton novekszik / legalabb egyszer csokkent szigoruan
vector < vector < vector <ULL> > > dp;

string s;
int n;

int main()
{
	cin >> s;
	n = s.SZ;
	s = "#" + s;
	dp.RS(n + 1, vector < vector <ULL> > (10, vector <ULL> (2)));
	FOR(j, 0, 9) dp[1][j][0] = 1;
	
	FOR(i, 1, n)
		FOR(j, 0, 9)
		{
			FOR(d, 0, j) dp[i][j][0] += dp[i - 1][d][0];
			FOR(d, j, 9) dp[i][j][1] += dp[i - 1][d][1];
			FOR(d, j + 1, 9) dp[i][j][1] += dp[i - 1][d][0];
		}

	ULL sum = 0;
	int digit = s[1] - '0';
	VI a = {0, digit};
	FOR(d, 0, digit - 1) sum += dp[n][d][0] + dp[n][d][1];
	FOR(i, 2, n)
	{
		digit = s[i] - '0';
		a.PB(digit);

		FOR(d, 0, digit - 1)
		{
			a[i] = d;
			bool inc = false, dec = false, ok = true;
			FOR(j, 2, i)
				if (a[j - 1] < a[j]) 
					if (dec) 
					{
						ok = false;
						break;
					}
					else inc = true;
				else
					if (a[j - 1] > a[j]) dec = true;
			if (!ok) continue;
			sum += dp[n + 1 - i][d][0];
			if (!dec) sum += dp[n + 1 - i][d][1];
		}

		a[i] = digit;
	}
	cout << sum;

	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 45/45
1 Elfogadva 0/0 3ms 1816 KiB
2 Elfogadva 0/0 3ms 2060 KiB
3 Elfogadva 0/0 3ms 2304 KiB
4 Elfogadva 1/1 2ms 2352 KiB
5 Elfogadva 1/1 3ms 2600 KiB
6 Elfogadva 1/1 3ms 2708 KiB
7 Elfogadva 1/1 3ms 2944 KiB
8 Elfogadva 1/1 3ms 3128 KiB
9 Elfogadva 1/1 3ms 3188 KiB
10 Elfogadva 1/1 3ms 3320 KiB
11 Elfogadva 1/1 3ms 3436 KiB
12 Elfogadva 1/1 2ms 3432 KiB
13 Elfogadva 2/2 3ms 3440 KiB
14 Elfogadva 2/2 2ms 3444 KiB
15 Elfogadva 2/2 3ms 3552 KiB
16 Elfogadva 2/2 3ms 3796 KiB
17 Elfogadva 1/1 3ms 4008 KiB
18 Elfogadva 2/2 3ms 4116 KiB
19 Elfogadva 2/2 3ms 4184 KiB
20 Elfogadva 2/2 3ms 4204 KiB
21 Elfogadva 3/3 3ms 4464 KiB
22 Elfogadva 2/2 3ms 4500 KiB
23 Elfogadva 2/2 3ms 4448 KiB
24 Elfogadva 2/2 3ms 4408 KiB
25 Elfogadva 2/2 3ms 4548 KiB
26 Elfogadva 2/2 3ms 4556 KiB
27 Elfogadva 2/2 3ms 4456 KiB
28 Elfogadva 2/2 3ms 4668 KiB
29 Elfogadva 2/2 3ms 4460 KiB
30 Elfogadva 2/2 3ms 4608 KiB