Submission #1202786

#	Time	Username	Problem	Language	Result	Execution time	Memory
1202786		Acanikolic	Nice sequence (IZhO18_sequence)	C++20	76 / 100	2085 ms	55912 KiB

sequence

#include <bits/stdc++.h>

using namespace std;

const int N = 4e5 + 10;

const int inf = 1e9 + 7;

vector<int> g[N];
int n, m;

bool check(int x) {
  vector<int> in(x + 1);
  for(int i = min(n, m); i <= x; i++) {
    if(i >= n) {
      g[i].push_back(i - n);
      in[i - n]++;
    }
    if(i >= m) {
      g[i - m].push_back(i);
      in[i]++;
    }
  }
  queue<int> q;
  for(int i = 0; i <= x; i++) if(!in[i]) q.push(i);
  vector<int> order;
  while(!q.empty()) {
    int u = q.front();
    q.pop();
    order.push_back(u);
    for(int j : g[u]) {
      in[j]--;
      if(!in[j]) q.push(j);
    }
  }
  for(int i = 0; i <= x; i++) g[i].clear();
  return ((int)order.size() == x + 1);
}

vector<int> Solve(int x) {
  vector<int> in(x + 1);
  for(int i = min(n, m); i <= x; i++) {
    if(i >= n) {
      g[i].push_back(i - n);
      in[i - n]++;
    }
    if(i >= m) {
      g[i - m].push_back(i);
      in[i]++;
    }
  }
  queue<int> q;
  for(int i = 0; i <= x; i++) if(!in[i]) q.push(i);
  vector<int> order;
  while(!q.empty()) {
    int u = q.front();
    q.pop();
    order.push_back(u);
    for(int j : g[u]) {
      in[j]--;
      if(!in[j]) q.push(j);
    }
  }
  for(int i = 0; i <= x; i++) g[i].clear();
  vector<int> res(x + 1);
  int pos = 0;
  for(int i = 1; i <= x; i++) if(order[i] == 0) pos = i;
  for(int i = 0; i <= x; i++) res[order[i]] = i - pos;
  vector<int> ans(x);
  for(int i = 1; i <= x; i++) ans[i - 1] = res[i] - res[i - 1];
  return ans;
}

signed main() {
  ios_base::sync_with_stdio(false);
  cin.tie(NULL);

  int tt;
  cin >> tt;
  while(tt--) {
    cin >> n >> m;
    int l = 0, r = n + m - __gcd(n, m), ans = 0;
    while(l <= r) {
      int mid = l + r >> 1;
      if(check(mid)) {
        ans = mid;
        l = mid + 1;
      } else {
        r = mid - 1;
      }
    }
    cout << ans << "\n";
    if(!ans) continue;
    vector<int> sol = Solve(ans);
    for(int i = 0; i < ans; i++) cout << sol[i] << " ";
    cout << "\n";
  }
}

• 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