제출 #780477

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
780477		fatemetmhr	곤돌라 (IOI14_gondola)	C++17	100 / 100	31 ms	3144 KiB

gondola

//  ~ Be Name Khoda ~  //

#include "gondola.h"
#include <bits/stdc++.h>
//#pragma GCC optimize ("O3")
//#pragma GCC target("avx2")
//#pragma GCC optimize("unroll-loops,Ofast")

using namespace std;

typedef long long ll;

#define pb       push_back
#define mp       make_pair
#define all(x)   x.begin(), x.end()
#define fi       first
#define se       second

const int maxn  =  1e6   + 10;
const int maxn5 =  5e5   + 10;
const int maxnt =  1.2e6 + 10;
const int maxn3 =  1e3   + 10;
const int mod   =  1e9   + 9;
const ll  inf   =  1e18;

vector <int> av, num;


int valid(int n, int a[])
{
    for(int i =0 ; i < n; i++){
        a[i]--;
        num.pb(a[i]);
    }
    sort(all(num));
    num.resize(unique(all(num)) - num.begin());
    if(num.size() != n)
        return 0;
    for(int i = 0; i < n; i++) if(a[i] < n)
        av.pb(i);
    int mnid = 0;
    if(av.empty())
        return 1;
    for(int i = 0; i < av.size(); i++) if(a[av[i]] < a[av[mnid]])
        mnid = i;
    int keep = mnid;
    //cout << av.size() << endl;
    for(int i = 0; i < av.size(); i++){
        int nxt = mnid + 1;
        if(nxt == av.size())
            nxt = 0;
        int u = av[mnid], v = av[nxt];
        //cout << "here " << mnid << ' ' << nxt << ' ' << u << ' ' << v << endl;
        if((nxt != keep && a[u] > a[v]) || (n + a[v] - a[u]) % n != (v - u + n) % n)
            return 0;
        mnid = nxt;
    }
    return 1;
}

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

int replacement(int n, int a[], int ret[])
{   
    vector <pair<int, int>> av;
    for(int i = 0; i < n; i++)
        a[i]--;
    int mx = 0, base = -1;
    for(int i = 0; i < n; i++){
        if(a[i] > a[mx])
            mx = i;
        if(a[i] < n)
            base = i;
    }
    //cout << a[mx] << ' ' << n << endl;
    if(a[mx] == n - 1)
        return 0;
    for(int i = 0; i < n; i++) if(a[i] >= n && i != mx)
        av.pb({a[i], i});
    sort(all(av));
    int ind = 0;
    int pt = 0;
    int mxx = a[mx];
    a[mx] = (base == -1 ? mx : ((n + mx - base) + a[base]) % n);
    for(int x = n; x <= mxx; x++){
        if(ind < av.size() && av[ind].fi == x){
            ret[pt++] = 1 + (base == -1 ? av[ind].se : ((n + av[ind].se - base) + a[base]) % n);
            ind++;
        }
        else{
            ret[pt++] = 1 + a[mx];
            a[mx] = x;
        }
        //cout << pt << ' ' << x << endl;
    }
    return pt;
}

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

ll pw(ll a, ll b){
    ll ret = 1;
    for(; b; b >>= 1, a = a * a % mod)
        if(b & 1) ret = ret * a % mod;
    return ret;
}


int countReplacement(int n, int a[])
{
    if(!valid(n, a))
        return 0;
    bool ok = true;
    for(int i = 0; i < n; i++)
        ok &= (a[i] < n);
    if(ok)
        return 1;
    vector <pair<int, int>> av;
    for(int i = 0; i < n; i++) if(a[i] >= n)
        av.pb({a[i], i});
    ll rem = av.size(), ans = 1;
    if(av.size() == n)
        ans = n;
    sort(all(av));
    int ind = 0;
    int last = n;
    for(int i = 0; i < av.size(); i++){
        ans = ans * pw(int(av.size()) - i, av[i].fi - last) % mod;
        last = av[i].fi + 1;
    }
    return ans;

}

컴파일 시 표준 에러 (stderr) 메시지

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:37:19: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   37 |     if(num.size() != n)
      |        ~~~~~~~~~~~^~~~
gondola.cpp:44:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |     for(int i = 0; i < av.size(); i++) if(a[av[i]] < a[av[mnid]])
      |                    ~~^~~~~~~~~~~
gondola.cpp:48:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |     for(int i = 0; i < av.size(); i++){
      |                    ~~^~~~~~~~~~~
gondola.cpp:50:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |         if(nxt == av.size())
      |            ~~~~^~~~~~~~~~~~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:86:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   86 |         if(ind < av.size() && av[ind].fi == x){
      |            ~~~~^~~~~~~~~~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:122:18: warning: comparison of integer expressions of different signedness: 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  122 |     if(av.size() == n)
      |        ~~~~~~~~~~^~~~
gondola.cpp:127:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  127 |     for(int i = 0; i < av.size(); i++){
      |                    ~~^~~~~~~~~~~
gondola.cpp:121:8: warning: unused variable 'rem' [-Wunused-variable]
  121 |     ll rem = av.size(), ans = 1;
      |        ^~~
gondola.cpp:125:9: warning: unused variable 'ind' [-Wunused-variable]
  125 |     int ind = 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: 15 / 15
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10