Submission #1041236

#	Time	Username	Problem	Language	Result	Execution time	Memory
1041236		DeathIsAwe	Gondola (IOI14_gondola)	C++14	100 / 100	22 ms	6252 KiB

gondola

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

int valid(int n, int inputSeq[]) {
  unordered_set<int> dupcheck;
  for (int i=0;i<n;i++) {
    dupcheck.insert(inputSeq[i]);
  }
  if (dupcheck.size() < n) {
    return 0;
  }
  int start = -1, temp;
  for (int i=0;i<n;i++) {
    if (inputSeq[i] <= n) {
      start = i; 
      break;
    }
  }
  if (start == -1) {
    return 1;
  } else {
    for (int i=start+1;i<n;i++) {
      if (inputSeq[i] <= n) {
        temp = (inputSeq[i] - inputSeq[start] + n) % n;
        if (temp != i - start) {
          return 0;
        }
        start = i;
      }
    }
  }
  return 1;
}





int replacement(int n, int gondolaSeq[], int replacementSeq[]) {
  int temp;
  for (int i=0;i<n;i++) {
    if (gondolaSeq[i] <= n) {
      temp = gondolaSeq[i] - i;
      break;
    }
  }
  vector<pair<int,int>> bruh;
  for (int i=0;i<n;i++) {
    if (gondolaSeq[i] > n) {
      bruh.push_back(make_pair(gondolaSeq[i], i));
    }
  }
  sort(bruh.begin(), bruh.end());
  int tracker = 0;
  for (pair<int,int> i: bruh) {
    replacementSeq[tracker++] = (i.second + temp + n) % n;
    if (replacementSeq[tracker - 1] == 0) {
      replacementSeq[tracker - 1] = n;
    }
    while (tracker + n < i.first) {
      replacementSeq[tracker] = tracker + n;
      tracker++;
    }
  }
  return tracker;
}





ll powfunc(ll a, ll b) {
  if (b == 0) {
    return 1;
  }
  vector<bool> binvec;
  while (b > 0) {
    if (b % 2 == 0) {
      binvec.push_back(false);
    } else {
      binvec.push_back(true);
    }
    b /= 2;
  }
  binvec.pop_back();
  ll val = a;
  reverse(binvec.begin(), binvec.end());
  for (bool i: binvec) {
    val *= val;
    val %= 1000000009;
    if (i) {
      val *= a;
      val %= 1000000009;
    }
  }
  return val;
}


int countReplacement(int n, int inputSeq[]) {
  if (valid(n, inputSeq) == 0) {
    return 0;
  }
  vector<int> overs;
  for (int i=0;i<n;i++) {
    if (inputSeq[i] > n) {
      overs.push_back(inputSeq[i]);
    }
  }
  sort(overs.begin(),overs.end());
  int siz = overs.size(), below = n;
  ll ans = 1;
  bool bruh = false;
  if (siz == n) {
    bruh = true;
  }
  for (int i=0;i<siz;i++) {
    ans *= powfunc(siz - i, overs[i] - below - 1);
    ans %= 1000000009;
  
    if (bruh) {
      bruh = false;
      ans *= (ll)(siz - i);
      ans %= 1000000009;
    }
    below = overs[i];
  }
  return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:11:23: warning: comparison of integer expressions of different signedness: 'std::unordered_set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   11 |   if (dupcheck.size() < n) {
      |       ~~~~~~~~~~~~~~~~^~~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:58:43: warning: 'temp' may be used uninitialized in this function [-Wmaybe-uninitialized]
   58 |     replacementSeq[tracker++] = (i.second + temp + n) % n;
      |                                  ~~~~~~~~~^~~~~~

• 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