Submission #891546

#	Time	Username	Problem	Language	Result	Execution time	Memory
891546		AkibAzmain	Gondola (IOI14_gondola)	C++17	75 / 100	16 ms	2384 KiB

gondola

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

int valid(int n, int a[])
{
  bool y = true;
  for (int i = 1; y && i < n; ++i) y = (a[i] - a[i - 1] + n) % n == 1;
  return y;
}

//----------------------

int replacement(int n, int a[], int ans[])
{
  int org = -1;
  for (int i = 0; i < n; ++i)
    if (a[i] <= n)
      {
        int norg = (i - a[i] + 1 + n) % n;
        assert (org == -1 || org == norg);
        org = norg;
      }
  if (org == -1) org = 0;
  vector < pair < int, int > > v;
  for (int i = 0; i < n; ++i) if (a[i] > n) v.push_back ({ a[i], (i - org + n) % n + 1 });
  sort (v.begin (), v.end ());
  int i = 0;
  for (auto &x : v)
    {
      while (x.first != x.second)
        {
          ans[i] = x.second;
          x.second = i + n + 1;
          ++i;
        }
    }
  return i;
}

//----------------------
int countReplacement(int n, int a[])
{
  int org = -1;
  for (int i = 0; i < n; ++i)
    if (a[i] <= n)
      {
        int norg = (i - a[i] + 1 + n) % n;
        if (org != -1 && org != norg) return 0;
        org = norg;
      }
  bool ind = false;
  if (org == -1) org = 0, ind = true;
  vector < ll > v;
  for (int i = 0; i < n; ++i) if (a[i] > n) v.push_back (a[i]);
  sort (v.begin (), v.end ());
  ll vs = v.size (), ans = 1, p = n, MOD = 1000000009;
  for (auto &x : v)
    {
      ll m = 1;
      for (int i = 0; i < x - p - 1; ++i) (m *= vs) %= MOD;
      (ans *= m) %= MOD;
      --vs;
      p = x;
    }
  return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:53:8: warning: variable 'ind' set but not used [-Wunused-but-set-variable]
   53 |   bool ind = false;
      |        ^~~

• 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