Submission #967159

#	Time	Username	Problem	Language	Result	Execution time	Memory
967159		VMaksimoski008	Gondola (IOI14_gondola)	C++17	75 / 100	30 ms	4956 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>

using namespace std;
using ll = long long;

int valid(int n, int inputSeq[]) {
    int ans = 1;
    int cnt = 0;
    set<int> s;

    int big = 0;
    for(int i=0; i<n; i++) {
      if(inputSeq[i] > n) big++;
    }

    vector<int> v(n);
    int pos = -1;
    for(int i=0; i<n; i++) {
      if(inputSeq[i] <= n) {
        pos = i;
        v[i] = inputSeq[i];
        break;
      }
    }

    for(int i=pos+1; i<n; i++) {
      v[i] = v[i-1] + 1;
      //if(v[i] > n) v[i] -= n;
    }

    for(int i=pos-1; i>=0; i--) {
      v[i] = v[i+1] - 1;
      //if(v[i] <= 0) 
    }

    for(int i=0; i<n; i++) {
      if(v[i] > n) v[i] -= n;
      else if(v[i] <= 0) v[i] += n;
    }

    for(int i=0; i<n; i++) {
      if(inputSeq[i] > n) continue;
      if(inputSeq[i] != v[i]) return 0;
    }

    for(int i=0; i<n; i++) s.insert(inputSeq[i]);
    if(s.size() < n) return 0;

    return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[]) {
  int cnt = 0, big = 0, arr[n];
  for(int i=0; i<n; i++) arr[i] = gondolaSeq[i];

  for(int i=0; i<n; i++) big += (arr[i] > n);

  if(big == n) {
    vector<pair<int, int> > v;
    for(int i=0; i<n; i++) v.push_back({ arr[i], i + 1 });
    sort(v.begin(), v.end());

    int curr = n + 1;
    for(int i=0; i<n; i++) {
      int a = v[i].second, b = v[i].first;
      replacementSeq[cnt++] = a;
      while(curr != b) {
        replacementSeq[cnt++] = curr++;
      }
      curr++;
    }
  } else {
    vector<int> v(n);
    int pos = -1;
    for(int i=0; i<n; i++) {
      if(arr[i] <= n) {
        pos = i;
        v[i] = arr[i];
        break;
      }
    }

    for(int i=pos+1; i<n; i++) v[i] = v[i-1] + 1;
    for(int i=pos-1; i>=0; i--) v[i] = v[i+1] - 1;

    for(int i=0; i<n; i++) {
      if(v[i] > n) v[i] -= n;
      else if(v[i] <= 0) v[i] += n;
    }

    vector<pair<int, int> > ord;
    for(int i=0; i<n; i++) 
      if(arr[i] > n) ord.push_back({ arr[i], v[i] });
    sort(ord.begin(), ord.end());

    int curr = n + 1;
    for(auto &[b, a] : ord) {
      replacementSeq[cnt++] = a;
      while(curr != b) replacementSeq[cnt++] = curr++;
      curr++;
    }
  }

  return cnt;
}

int countReplacement(int n, int inputSeq[]) {
  if(!valid(n, inputSeq)) return 0;
  const int mod = 1e9 + 9;
  ll ans = 1;

  vector<int> v;
  for(int i=0; i<n; i++)
    if(inputSeq[i] > n) v.push_back(inputSeq[i]);
  if(v.size() == 0) return 1;
  sort(v.begin(), v.end());

  int mx = v.back();
  int sz = v.size();
  for(int i=n+1; i<=mx; i++) {
    int p = lower_bound(v.begin(), v.end(), i) - v.begin();
    if(v[p] == i) continue;
    ans = (ans * (sz - p)) % mod;
  }

  return (int)ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:48:17: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |     if(s.size() < n) return 0;
      |        ~~~~~~~~~^~~
gondola.cpp:8:9: warning: unused variable 'ans' [-Wunused-variable]
    8 |     int ans = 1;
      |         ^~~
gondola.cpp:9:9: warning: unused variable 'cnt' [-Wunused-variable]
    9 |     int cnt = 0;
      |         ^~~

• 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: 0 / 15
• Subtask 10: 0 / 10