4906 2023. 04. 06 16:27:52 SleepyOverlord Következő permutáció (50 pont) cpp17 Elfogadva 50/50 23ms 5952 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

#define IN_FILE "a.in"
#define OUT_FILE "a.out"

int n;
VI a, biggestBefore, tree, tree2;

int prefMax(int node, int left, int right, int l, int r, VI &t)
{
	if (l > r) return 0;
	if (l <= left && right <= r) return t[node];
	int mid = (left + right) / 2;
	int ans = 0;
	if (l <= mid) ans = prefMax(node * 2, left, mid, l, r, t);
	if (r > mid) ans = max(ans, prefMax(node * 2 + 1, mid + 1, right, l, r, t));
	return ans;
}

void update(int node, int left, int right, int index, int value, VI &t)
{
	if (left == right)
	{
		t[node] = value;
		return;
	}
	int mid = (left + right) / 2;
	if (index <= mid) update(node * 2, left, mid, index, value, t);
	else update(node * 2 + 1, mid + 1, right, index, value, t);
	t[node] = max(t[node * 2], t[node * 2 + 1]);
}

bool good(int value, int index, SI &unused)
{
	bool ok1 = prefMax(1, 1, n, 1, value - 1, tree2) <= value;
	auto it = unused.upper_bound(value);
	bool ok2 = it == unused.EN || *it > prefMax(1, 1, n, 1, index - 1, tree);
	return ok1 && ok2;
}

int main()
{
	cin >> n;
	a.RS(n + 1);
	tree.RS(4 * n + 1); //max(a[i..j])
	tree2.RS(4 * n + 1); //max(biggestBefore[i..j])
	biggestBefore.RS(n + 1);
	FOR(i, 1, n)
	{
		cin >> a[i];
		update(1, 1, n, i, a[i], tree);
		biggestBefore[a[i]] = prefMax(1, 1, n, 1, i - 1, tree);
		update(1, 1, n, a[i], biggestBefore[a[i]], tree2);
	}

	SI unused;
	int i = n;
	while (1)
	{
		unused.INS(a[i]);
		biggestBefore[a[i]] = 0;
		update(1, 1, n, a[i], biggestBefore[a[i]], tree2);
		auto it = unused.upper_bound(a[i]);
		if (it != unused.EN)
		{
			if (good(*it, i, unused))
			{
				a[i] = *it;
				update(1, 1, n, i, a[i], tree);
				biggestBefore[a[i]] = prefMax(1, 1, n, 1, i - 1, tree);
				update(1, 1, n, a[i], biggestBefore[a[i]], tree2);
				unused.erase(it);
				int firstBad = -1;
				FOR(j, i + 1, n)
				{
					int aux = prefMax(1, 1, n, 1, j - 1, tree);
					auto it2 = unused.lower_bound(aux);
					if (it2 != unused.BG) --it2;
					bool ok = false;
					while (it2 != unused.EN)
					{
						if (good(*it2, j, unused))
						{
							a[j] = *it2;
							update(1, 1, n, j, a[j], tree);
							biggestBefore[a[j]] = prefMax(1, 1, n, 1, j - 1, tree);
							update(1, 1, n, a[j], biggestBefore[a[j]], tree2);
							unused.erase(it2);
							ok = true;
							break;
						}
						++it2;
					}
					if (!ok)
					{
						firstBad = j;
						break;
					}
				}
				if (firstBad == -1)
				{
					FOR(i, 1, n) cout << a[i] << " ";
					cout << endl;
					return 0;
				}
				else 
				{
					FOR(j, i + 1, firstBad - 1) 
					{
						unused.INS(a[j]);
						biggestBefore[a[j]] = 0;
						update(1, 1, n, a[j], biggestBefore[a[j]], tree2);
					}
					i = firstBad - 1;
				}
			}
			else
			{
				biggestBefore[a[i]] = 0;
				update(1, 1, n, a[i], biggestBefore[a[i]], tree2);
				--i;
			}
		}
		else 
		{
			biggestBefore[a[i]] = 0;
			update(1, 1, n, a[i], biggestBefore[a[i]], tree2);
			--i;
		}
	}

	cout << -1;
	return 0;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 50/50
1 Elfogadva 0/0 3ms 1812 KiB
2 Elfogadva 0/0 7ms 2552 KiB
3 Elfogadva 1/1 3ms 2396 KiB
4 Elfogadva 1/1 2ms 2492 KiB
5 Elfogadva 1/1 2ms 2700 KiB
6 Elfogadva 1/1 2ms 2780 KiB
7 Elfogadva 2/2 2ms 2912 KiB
8 Elfogadva 2/2 2ms 3120 KiB
9 Elfogadva 2/2 2ms 3368 KiB
10 Elfogadva 2/2 2ms 3488 KiB
11 Elfogadva 1/1 2ms 3580 KiB
12 Elfogadva 1/1 2ms 3560 KiB
13 Elfogadva 1/1 2ms 3560 KiB
14 Elfogadva 1/1 3ms 3800 KiB
15 Elfogadva 3/3 3ms 3860 KiB
16 Elfogadva 3/3 3ms 4100 KiB
17 Elfogadva 3/3 3ms 4340 KiB
18 Elfogadva 3/3 4ms 4336 KiB
19 Elfogadva 3/3 8ms 4792 KiB
20 Elfogadva 3/3 8ms 4824 KiB
21 Elfogadva 3/3 9ms 5016 KiB
22 Elfogadva 3/3 10ms 5016 KiB
23 Elfogadva 3/3 10ms 5108 KiB
24 Elfogadva 3/3 10ms 5020 KiB
25 Elfogadva 2/2 23ms 5952 KiB
26 Elfogadva 2/2 10ms 5220 KiB