Submission #239512

#	Time	Username	Problem	Language	Result	Execution time	Memory
239512		Autoratch	Gondola (IOI14_gondola)	C++14	60 / 100	57 ms	10872 KiB

gondola

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

const int N = 3e5 + 1;
const int MOD = 1e9 + 9;

int repl[N];
set<int> s;

int valid(int n,int a[])
{
    int x,y = 0;
    for(int i = 0;i < n;i++) s.insert(a[i]);
    if(s.size()<n) return 0;
    for(int i = 0;i < n;i++) if(a[i]<=n) x = i,y = a[i];
    if(y==0) return 1;
    for(int i = 0;i < n;i++) if(a[i]<=n)
    {
        int z = y+i-x;
        if(z>n) z%=n;
        if(z<1) z+=n;
        if(z!=a[i]) return 0;
    }
    return 1;
}

int replacement(int n,int a[],int ans[])
{
    int x,y = 0,mx = 0;
    for(int i = 0;i < n;i++) 
    {
        if(a[i]<=n) x = i,y = a[i];
        mx = max(mx,a[i]);
    }
    int ret = mx-n;
    int bx = mx;
    for(int i = 0;i < n;i++)
    {
        int z = y+i-x;
        if(z>n) z%=n;
        if(z<1) z+=n;
        if(z!=a[i]) repl[a[i]] = z;
    }
    set<int> ss;
    for(int i = 0;i < ret;i++)
    {
        int j = i+n+1;
        if(!repl[j]) repl[j] = repl[bx],repl[bx] = j;
        ans[i] = repl[j];
        ss.insert(ans[i]);
    }
    assert(ss.size()==ret);
    return ret;
}

int countReplacement(int n,int a[])
{
    int x,y = 0,mx = 0;
    for(int i = 0;i < n;i++) 
    {
        if(a[i]<=n) x = i,y = a[i];
        mx = max(mx,a[i]);
        s.insert(a[i]);
    }
    if(s.size()<n) return 0;
    for(int i = 0;i < n;i++)
    {
        int z = y+i-x;
        if(z>n) z%=n;
        if(z<1) z+=n;
        if(z!=a[i] and a[i]<=n) return 0;
        if(z!=a[i]) repl[a[i]] = z;
    }
    int s = 0,t = 0;
    for(int i = n+1;i <= mx;i++) if(!repl[i]) s++; else t++;
    long long res = 1;
    while(s--) res = (res*t)%MOD;
    int ans = res;
    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:15:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(s.size()<n) return 0;
        ~~~~~~~~^~
In file included from /usr/include/c++/7/cassert:44:0,
                 from /usr/include/x86_64-linux-gnu/c++/7/bits/stdc++.h:33,
                 from gondola.cpp:2:
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:53:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     assert(ss.size()==ret);
            ~~~~~~~~~^~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:66:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(s.size()<n) return 0;
        ~~~~~~~~^~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:30:9: warning: 'x' may be used uninitialized in this function [-Wmaybe-uninitialized]
     int x,y = 0,mx = 0;
         ^
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:59:9: warning: 'x' may be used uninitialized in this function [-Wmaybe-uninitialized]
     int x,y = 0,mx = 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: 0 / 15
• Subtask 9: 0 / 15
• Subtask 10: 0 / 10