Submission #707138

#	Time	Username	Problem	Language	Result	Execution time	Memory
707138		Alihan_8	Gondola (IOI14_gondola)	C++17	90 / 100	20 ms	2768 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h"
// include <ext/pb_ds/assoc_container.hpp>
// include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
using namespace std;
#define all(x) x.begin(), x.end()
#define pb push_back
// define ordered_set tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>
#define mpr make_pair
#define ln '\n'
void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
//#define int long long
int valid(int n, int inputSeq[]){
    vector <pair<int,int>> res;
    for ( int i = 0; i < n; i++ ){
        if ( inputSeq[i] <= n ) res.pb({inputSeq[i], i});
    }
    sort(all(res));
    for ( int i = 1; i < (int)res.size(); i++ ){
        auto [x, y] = res[i-1];
        auto [l, r] = res[i];
        r = (r+n-y)%n;
        if ( l-x != r ) return false;
    }
    return true;
}
int replacement(int n, int gondolaSeq[], int ans[]){
    vector <int> p;
    for ( int i = 0; i < n; i++ ){
        p.pb(gondolaSeq[i]);
    }
    int it = -1;
    for ( int i = 0; i < n; i++ ){
        if ( p[i] <= n ) it = i;
    }
    auto F = [&](int x){ return x%n == 0 ? n : x%n; };
    vector <int> res(n);
    if ( it == -1 ) iota(all(res), 1);
    else{
        for ( int j = it; j < n; j++ ) res[j] = F(p[it]+j-it);
        for ( int j = 0; j < it; j++ ) res[j] = F(p[it]+n-it+j);
    }
    vector <pair<int,int>> cur;
    for ( int i = 0; i < n; i++ ){
        cur.pb({p[i], i});
    }
    sort(all(cur));
    int  _i = 0, last = n+1;
    for ( auto [val, it]: cur ){
        while ( res[it] < val ){
            ans[_i++] = res[it];
            res[it] = last++;
        }
    }
    return _i;
}
typedef long long ll;
int countReplacement(int n, int inputSeq[]){
    if ( !valid(n, inputSeq) ) return 0;
    const ll Mod = 1e9+9;
    vector <int> p;
    for ( int i = 0; i < n; i++ ){
        if ( inputSeq[i] > n ) p.pb(inputSeq[i]);
    }
    ll cnt = 1;
    if ( (int)p.size() == n ) cnt = n;
    auto bin_pow = [&](ll x, int y){
        ll res = 1;
        while ( y ){
            if ( y & 1 ) res = res*x%Mod;
            x = x*x%Mod;
            y >>= 1;
        }
        return res;
    };
    sort(all(p));
    int last = n+1, sz = (int)p.size();
    for ( int i = 0; i < sz; i++ ){
        cnt = cnt*bin_pow(sz-i, p[i]-last)%Mod;
        last = p[i]+1;
    }
    return cnt;
}
#if false
signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int sub_task, n; cin >> n;
    int p[n] = {0};
    for ( int i = 0; i < n; i++ ) cin >> p[i];
    cout << countReplacement(n, p);

    cout << '\n';
/*
4
1 2 7 6
answer: 2
7
2 3 4 12 6 7 1
answer: 1
2
3 4
answer: 2
*/
}
#endif

Compilation message (stderr)

gondola.cpp: In function 'void IO(std::string)':
gondola.cpp:12:29: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   12 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
gondola.cpp:12:70: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   12 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                                                               ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 0 / 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