Submission #488601

#	Time	Username	Problem	Language	Result	Execution time	Memory
488601		keta_tsimakuridze	Nice sequence (IZhO18_sequence)	C++14	76 / 100	2065 ms	35936 KiB

sequence

#include<bits/stdc++.h>
#define f first
#define s second
#define pii pair<int,int>
//#define int long long
using namespace std;
const int N = 4e5 + 5, mod = 1e9 + 7; // !
int t, in[N],n,m, p[N];
vector<int> V[N], v;
queue<int> q;
int check(int x) {
	for(int i = 0; i <= x; i++) in[i] = 0, V[i].clear();
	for(int i = 0; i <= x; i++) {
		if(i + n <= x) {
			in[i + n]++;
			V[i].push_back(i + n);
		}
		if(i + m <= x) {
			in[i]++;
			V[i + m].push_back(i);
		}
	}
	for(int i = 0; i <= x; i++) {
		if(!in[i]) q.push(i);
	}	
	v.clear();
	while(q.size()) {
		int u = q.front();
		q.pop();
		v.push_back(u);
		for(int i = 0; i < V[u].size(); i++) {
			in[V[u][i]]--;
			if(!in[V[u][i]]) q.push(V[u][i]);
		}
	}
	return v.size();
}
main() {
	ios_base::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	#define endl " "
	cin >> t;
	while(t--) {
		cin >> n >> m;
		int l = max(n, m), r = 400000, ans = max(n, m) - 1;
		while(l <= r) {
			int mid = (l + r)/2;
			if(check(mid) == mid + 1) {
				ans = mid, l = mid + 1;
			} else r = mid - 1;
		}

		cout << ans << endl;
		check(ans);
		reverse(v.begin(), v.end());
		for(int i = 0; i < v.size(); i++) {
			p[v[i]] = i + 1;
		}
		for(int i = 1; i <= ans; i++) {
			cout << p[i] - p[i - 1] <<" ";
	
		}
		cout << endl;
	}
}

Compilation message (stderr)

sequence.cpp: In function 'int check(int)':
sequence.cpp:31:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |   for(int i = 0; i < V[u].size(); i++) {
      |                  ~~^~~~~~~~~~~~~
sequence.cpp: At global scope:
sequence.cpp:38:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   38 | main() {
      | ^~~~
sequence.cpp: In function 'int main()':
sequence.cpp:55:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |   for(int i = 0; i < v.size(); i++) {
      |                  ~~^~~~~~~~~~

• Subtask 1: 6 / 6
• Subtask 2: 9 / 9
• Subtask 3: 14 / 14
• Subtask 4: 15 / 15
• Subtask 5: 14 / 14
• Subtask 6: 18 / 18
• Subtask 7: 0 / 24