Submission #1166732

#	Time	Username	Problem	Language	Result	Execution time	Memory
1166732		HappyCapybara	Gondola (IOI14_gondola)	C++17	100 / 100	28 ms	5464 KiB

gondola

#include "gondola.h"
#include<bits/stdc++.h>
using namespace std;

#define ll long long

ll m = 1000000009;

int valid(int n, int inputSeq[]){
  int x = -1;
  unordered_set<int> s;
  for (int i=0; i<n; i++){
    if (s.find(inputSeq[i]) != s.end()) return 0;
    s.insert(inputSeq[i]);
    if (inputSeq[i] <= n){
      if (x != -1 && x != inputSeq[i]) return 0;
      if (x == -1) x = inputSeq[i];
    }
    if (x != -1) x = (x % n) + 1;
  }
  return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[]){
  vector<ll> v(n);
  ll ssf = (ll) pow(10, 10);
  int si;
  for (int i=0; i<n; i++){
    if (gondolaSeq[i] < ssf){
      ssf = gondolaSeq[i];
      si = i;
    }
  }
  if (ssf > n){
    ssf = 1;
    si = 0;
  }
  priority_queue<pair<int,int>> pq;
  for (int i=0; i<n; i++){
    v[(ssf+i-1)%n] = gondolaSeq[(si+i)%n];
    if (v[(ssf+i-1)%n] > n) pq.push({-v[(ssf+i-1)%n], (ssf+i-1)%n});
  }
  int cur = n+1;
  while (!pq.empty()){
    int x = -pq.top().first, i = pq.top().second;
    pq.pop();
    bool nw = true;
    while (cur <= x){
      if (nw) replacementSeq[cur-n-1] = i+1;
      else replacementSeq[cur-n-1] = cur-1;
      nw = false;
      cur++;
    }
  }
  return cur-n-1;
}

ll power(ll a, ll b){
  if (b == 0) return 1;
  if (b == 1) return a;
  ll res = power(a, b/2);
  res = ((ll)res*(ll)res)%m;
  if (b % 2) return (a*res) % m;
  return res;
}

int countReplacement(int n, int inputSeq[]){
  if (!valid(n, inputSeq)) return 0;
  ll res = 1, r = 0, cur = n+1;
  priority_queue<ll> pq;
  for (int i=0; i<n; i++){
    if (inputSeq[i] > n){
      r++;
      pq.push(-inputSeq[i]);
    }
  }
  if (r == n) res = n;
  while (!pq.empty()){
    ll next = -pq.top();
    pq.pop();
    res = (res*power(r, next-cur)) % m;
    cur = next+1;
    r--;
  }
  return res;
}

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 10 / 10
• Subtask 4: 5 / 5
• Subtask 5: 10 / 10
• Subtask 6: 20 / 20
• Subtask 7: 5 / 5
• Subtask 8: 15 / 15
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10