Submission #1042930

#	Time	Username	Problem	Language	Result	Execution time	Memory
1042930		dpsaveslives	Gondola (IOI14_gondola)	C++17	75 / 100	28 ms	5204 KiB

gondola

#include "gondola.h"
#include <bits/stdc++.h>
#define f first
#define s second
#define ll long long
using namespace std;
const int MAXN = 1e5;
const int MOD = 1e9+9;
ll exp(ll x, ll n){
    x %= MOD;
    ll res = 1;
    while(n > 0){
        if(n % 2 == 1){
            res = res*x%MOD;
        }
        x = x*x%MOD;
        n /= 2;
    }
    return res;
}
int valid(int n, int inputSeq[]){
    int pos = -1; set<int> vals;
    for(int i = 0;i<n;++i){
        if(inputSeq[i] <= n && pos == -1){
            pos = i;
        }
        vals.insert(inputSeq[i]);
    }
    if(vals.size() != n){
        return 0;
    }
    if(pos == -1){
        return 1;
    }
    int cur = inputSeq[pos];
    for(int i = pos;i<n;++i){
        if(inputSeq[i] <= n){
            //cout << i << " " << cur << "\n";
            if(inputSeq[i] != cur){
                return 0;
            }
        }
        if(cur == n){
            cur = 0;
        }
        ++cur;
    }
    for(int i = 0;i<=pos-1;++i){
        if(inputSeq[i] <= n){
            if(inputSeq[i] != cur){
                return 0;
            }
        }
        if(cur == n){
            cur = 0;
        }
        ++cur;

    }
    return 1;
}
int replacement(int n, int gondolaSeq[], int replacementSeq[]){
    vector<pair<int,int>> arr;
    int pos = -1; bool any = false;
    for(int i = 0;i<n;++i){
        if(gondolaSeq[i] <= n){
            if(pos == -1){
                pos = i;
            }
        }
        else{
            any = true;
            arr.push_back({gondolaSeq[i],i});
        }
    }
    if(!any){
        return 0;
    }
    int rep = 0, cur;
    if(pos != -1){
        cur = gondolaSeq[pos];
    }
    else{
        cur = 1;
        pos = 0;
    }
    vector<int> reps(n,-1);
    for(int i = pos;i<n;++i){
        if(gondolaSeq[i] > n){
            reps[i] = cur;
        }
        if(cur == n){
            cur = 0;
        }
        ++cur;
    }
    for(int i = 0;i<=pos-1;++i){
        if(gondolaSeq[i] > n){
            reps[i] = cur;
        }
        if(cur == n){
            cur = 0;
        }
        ++cur;
    }
    sort(arr.begin(),arr.end());
    cur = n+1; int i = 0;
    while(i < arr.size()){
        replacementSeq[rep] = reps[arr[i].s];
        //cout << rep << " " << replacementSeq[rep] << "\n";
        ++rep;
        while(cur < arr[i].f){ //until arr[i] = cur
            replacementSeq[rep] = cur;
            //cout << rep << " " << replacementSeq[rep] << "\n";
            ++rep; ++cur;
        }
        ++cur; ++i;
    }
    return rep;
}
int countReplacement(int n,	int	inputSeq[]){
    if(!valid(n,inputSeq)) return 0;
    int pos = -1; bool any = false;
    vector<pair<int,int>> arr;
    for(int i = 0;i<n;++i){
        if(inputSeq[i] <= n){
            if(pos == -1){
                pos = i;
            }
        }
        else{
            any = true;
            arr.push_back({inputSeq[i],i});
        }
    }
    if(!any){
        return 1;
    }
    ll ans = 1ll;
    if(pos == -1){ //no value that is <= n
        ans = n;
    }
    /*sort(arr.begin(),arr.end());
    int cur = n; int i = 0; ll cnt = arr.size();
    while(i < arr.size()){
        ll x = exp(1ll*cnt,1ll*(arr[i].f-cur));
        ans = ans*x%MOD;
        cur = arr[i].f;
        ++i; --cnt;
    }*/
    if(pos != -1){
        sort(arr.begin(),arr.end());
        int cur = n; int i = 0; ll cnt = arr.size();
        while(i < arr.size()){
            ll x = exp(1ll*cnt,1ll*(arr[i].f-cur-1));
            ans = ans*x%MOD;
            cur = arr[i].f;
            ++i; --cnt;
        }
    }

    return ans;
}

Compilation message (stderr)

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:29:20: warning: comparison of integer expressions of different signedness: 'std::set<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   29 |     if(vals.size() != n){
      |        ~~~~~~~~~~~~^~~~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:108:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  108 |     while(i < arr.size()){
      |           ~~^~~~~~~~~~~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:154:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  154 |         while(i < arr.size()){
      |               ~~^~~~~~~~~~~~

• 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