답안 #1042930

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1042930 2024-08-03T14:58:17 Z dpsaveslives 곤돌라 (IOI14_gondola) C++17
75 / 100
28 ms 5204 KB
#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

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()){
      |               ~~^~~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 444 KB Output is correct
4 Correct 0 ms 440 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 1 ms 348 KB Output is correct
6 Correct 6 ms 2396 KB Output is correct
7 Correct 17 ms 4188 KB Output is correct
8 Correct 11 ms 4356 KB Output is correct
9 Correct 4 ms 1736 KB Output is correct
10 Correct 16 ms 4956 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 7 ms 2396 KB Output is correct
7 Correct 17 ms 4312 KB Output is correct
8 Correct 11 ms 4444 KB Output is correct
9 Correct 4 ms 1628 KB Output is correct
10 Correct 15 ms 4956 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 10 ms 2396 KB Output is correct
14 Correct 0 ms 400 KB Output is correct
15 Correct 21 ms 5204 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 420 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 0 ms 448 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 440 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 444 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 448 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 1 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 4 ms 1116 KB Output is correct
12 Correct 5 ms 1116 KB Output is correct
13 Correct 7 ms 1492 KB Output is correct
14 Correct 4 ms 1080 KB Output is correct
15 Correct 12 ms 2384 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 424 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 344 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 21 ms 4664 KB Output is correct
10 Correct 18 ms 3676 KB Output is correct
11 Correct 6 ms 1628 KB Output is correct
12 Correct 8 ms 2004 KB Output is correct
13 Incorrect 1 ms 832 KB Output isn't correct
14 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 440 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 28 ms 4480 KB Output is correct
10 Correct 19 ms 3744 KB Output is correct
11 Correct 6 ms 1628 KB Output is correct
12 Correct 8 ms 1916 KB Output is correct
13 Incorrect 1 ms 604 KB Output isn't correct
14 Halted 0 ms 0 KB -