Submission #1017840

#	Time	Username	Problem	Language	Result	Execution time	Memory
1017840		serkanrashid	Gondola (IOI14_gondola)	C++14	100 / 100	29 ms	6764 KiB

gondola

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

using namespace std;

const long long mod = 1e9+9;
const int maxn = 1e5+5;

int n,a[maxn];

bool check_valid()
{
    unordered_set<int>used;
    int st = -1;

    for(int i = 0; i < n; i++)
    {
        if(used.find(a[i])!=used.end()) return false;
        used.insert(a[i]);
        if(1<=a[i] && a[i] <= n) st = i;
    }
    if(st==-1) return true;

    int last = a[st];
    int i = (st+1)%n;

    while(i!=st)
    {
        last++;
        if(last>n) last = 1;
        if(1<=a[i] && a[i]<=n)
        {
            if(last!=a[i]) return false;
        }
        i = (i+1)%n;
    }
    return true;
}

int valid(int N, int inputSeq[])
{
    n = N;
    for(int i = 0; i < n; i++) a[i] = inputSeq[i];
    return check_valid();
}

int need[maxn];

bool make_need()
{
    int st = -1;
    for(int i = 0; i < n; i++) if(1 <= a[i] && a[i] <= n) st = i;

    if(st==-1)
    {
        for(int i = 0; i < n; i++) need[i] = i+1;
        return false;
    }
    int last = a[st];
    int i = (st+1)%n;
    while(i!=st)
    {
        last++;
        if(last>n) last = 1;
        need[i] = last;
        i = (i+1)%n;
    }
    return true;
}

vector<int> do_replace()
{
    int l = 0;
    for(int i = 0; i < n; i++) if(a[i]>l) l = a[i];
    l = l - n;

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

    vector<int>res;

    int last = n;
    for(pair<int,int> nb : pos)
    {
        res.push_back(nb.second);
        for(int j = last+1; j < nb.first; j++) res.push_back(j);
        last = nb.first;
    }
    return res;
}

int replacement(int N, int gondolaSeq[], int replacementSeq[])
{
    n = N;
    for(int i = 0; i < n; i++) a[i] = gondolaSeq[i];
    bool f = make_need();
    vector<int>pom = do_replace();
    for(int i = 0; i < pom.size(); i++) replacementSeq[i] = pom[i];
    return pom.size();
}

long long step(long long a, long long b)
{
    long long res = 1;
    while(b>0)
    {
        if(b%2) res = (res*a)%mod;
        a = (a*a)%mod;
        b /= 2;
    }
    return res;
}

int countReplacement(int N, int inputSeq[])
{
    n = N;
    for(int i = 0; i < n; i++) a[i] = inputSeq[i];
    if(!check_valid()) return 0;
    ///OSHTE

    vector<int>b;
    for(int i = 0; i < n; i++) if(a[i]>n) b.push_back(a[i]);
    sort(b.begin(),b.end());

    long long ans = 1;
    bool f = make_need();
    if(!f) ans = n;

    int last = n;
    for(int i = 0; i < b.size(); i++)
    {
        int st = b[i] - last - 1;
        int osn = b.size()-i;
        ans = (ans*step(osn,st))%mod;
        last = b[i];
    }
    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:99:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   99 |     for(int i = 0; i < pom.size(); i++) replacementSeq[i] = pom[i];
      |                    ~~^~~~~~~~~~~~
gondola.cpp:97:10: warning: unused variable 'f' [-Wunused-variable]
   97 |     bool f = make_need();
      |          ^
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:131:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  131 |     for(int i = 0; i < b.size(); i++)
      |                    ~~^~~~~~~~~~

• 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