제출 #249840

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
249840		ant101	곤돌라 (IOI14_gondola)	C++14	100 / 100	69 ms	6008 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h" 
using namespace std;
 
#define F first
#define S second
 
const int N = 100005;
const int MOD = 1000000009;
 
int valid(int n, int A[]) {
    set <int> S(A, A + n);
    if (S.size() != n) return 0;
    for (int i = 0, k = -MOD; i < n; i++) {
        if (A[i] <= n) {
            if (k == -MOD) k = (A[i] - i - 1 + n) % n;
            else if (k != (A[i] - i - 1 + n) % n) return 0;
        }
    }
    return 1;
}
 
int replacement(int n, int A[], int ret[]) {
    int k = -MOD, ptr = 0;
    vector <int> B(N);
    for (int i = 0; i < n; i++) {
        if (A[i] <= n) {
            k = (A[i] - i - 1 + n) % n;
        }
    }
    vector <pair <int, int>> v;
    for(int i = 0; i < n; i++) {
        if (k == -MOD) B[i] = A[i];
        else B[i] = A[(i - k + n) % n];
        if (B[i] > n) v.push_back({B[i], i + 1});
    }
    sort(v.begin(), v.end());
    for (int i = 0; i < v.size(); i++) {
        while (ptr + n < v[i].F) {
            ret[ptr++] = v[i].S;
            v[i].S = ptr + n;
        }
    }
    return ptr;
}
 
int modpow(int a, int b) {
    int ret = 1;
    for (; b; b >>= 1) {
        if (b & 1) ret = 1LL * ret * a % MOD;
        a = 1LL * a * a % MOD;
    }
    return ret;
}
 
int countReplacement(int n, int A[]) {
    if (!valid(n, A)) return 0;
    vector <int> v;
    for (int i = 0; i < n; i++) if(A[i] > n) v.push_back(A[i]);
    sort (v.begin(), v.end());
    int ans = 1;
    for (int i = 0; i < v.size(); i++) {
        int b = v.size() - i;
        int p = v[i] - (i ? v[i - 1] + 1 : n + 1);
        ans = 1LL * ans * modpow(b, p) % MOD;
    }
    if (v.size() == n) ans = 1LL * ans * n % MOD;
    return ans;
}

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

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:13:18: 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:38:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < v.size(); i++) {
                     ~~^~~~~~~~~~
gondola.cpp: In function 'int countReplacement(int, int*)':
gondola.cpp:62:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < v.size(); i++) {
                     ~~^~~~~~~~~~
gondola.cpp:67:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if (v.size() == n) ans = 1LL * ans * n % MOD;
         ~~~~~~~~~^~~~

• 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