제출 #1358404

#제출 시각아이디문제언어결과실행 시간메모리
1358404Charizard2021Gondola (IOI14_gondola)C++17
60 / 100
24 ms5820 KiB
#include "gondola.h"
#include<bits/stdc++.h>
using namespace std;
const int MOD = 1e9 + 9;
int valid(int n, int inputSeq[]){
    vector<int> seq(n);
    vector<int> seq2(2 * n);
    bool allGreat = true;
    set<int> s;
    for(int i = 0; i < n; i++){
        seq[i] = inputSeq[i];
        s.insert(seq[i]);
        if(seq[i] <= n){
            allGreat = false;
        }
    }
    if((int)s.size() != n){
        return 0;
    }
    for(int i = 0; i < n; i++){
        seq2[i + n] = seq[i];
        seq2[i] = seq[i];
    }
    if(allGreat){
        return 1;
    }
    else{
        bool works = true;
        for(int i = 0; i < n; i++){
            if(seq[i] <= n){
                int idx = 0;
                for(int j = 0; j < n; j++){
                    if(seq2[j + seq[i] - 1] == seq[i]){
                        idx = j;
                        break;
                    }
                }
                for(int x = idx; x < idx + n; x++){
                    if(seq2[x] <= n && seq2[x] != x - idx + 1){
                        works = false;
                        break;
                    }
                }
                break;
            }
        }   
        return (int)works;
    }
}
int replacement(int n, int gondolaSeq[], int replacementSeq[]){
    if(!valid(n, gondolaSeq)){
        return 0;
    }
    vector<int> seq(n);
    vector<int> seq2(2 * n);
    bool allGreat = true;
    for(int i = 0; i < n; i++){
        seq[i] = gondolaSeq[i];
        if(seq[i] <= n){
            allGreat = false;
        }
    }
    for(int i = 0; i < n; i++){
        seq2[i + n] = seq[i];
        seq2[i] = seq[i];
    }
    if(allGreat){
        vector<pair<int, int> > v;
        for(int i = 0; i < n; i++){
            if(seq[i] > n){
                v.push_back(make_pair(seq[i], i + 1));
            }
        }
        sort(v.begin(), v.end());
        int idx = 0;
        int prev = n;
        for(int i = 0; i < (int)v.size(); i++){
            replacementSeq[idx++] = v[i].second;
            prev++;
            for(int j = prev; j < v[i].first; j++){
                replacementSeq[idx++] = prev;
                prev++;
            }
        }
        return idx;
    }
    for(int i = 0; i < n; i++){
        if(seq[i] <= n){
            int idx = 0;
            for(int j = 0; j < n; j++){
                if(seq2[j + seq[i] - 1] == seq[i]){
                    idx = j;
                    break;
                }
            }
            vector<pair<int, int> > v;
            for(int i = idx; i < idx + n; i++){
                if(seq2[i] > n){
                    v.push_back(make_pair(seq2[i], i - idx + 1));
                }
            }
            sort(v.begin(), v.end());
            int idx2 = 0;
            int prev = n;
            for(int i = 0; i < (int)v.size(); i++){
                replacementSeq[idx2++] = v[i].second;
                prev++;
                for(int j = prev; j < v[i].first; j++){
                    replacementSeq[idx2++] = prev;
                    prev++;
                }
            }
            return idx2;
        }
    }
    return 0;
}
int countReplacement(int n, int inputSeq[]){
    if(!valid(n, inputSeq)){
        return 0;
    }
    vector<int> seq(n);
    vector<int> seq2(2 * n);
    bool allGreat = true;
    for(int i = 0; i < n; i++){
        seq[i] = inputSeq[i];
        if(seq[i] <= n){
            allGreat = false;
        }
    }
    for(int i = 0; i < n; i++){
        seq2[i + n] = seq[i];
        seq2[i] = seq[i];
    }
    if(allGreat){
        map<int, int> mp;
        int mxVal = 0;
        for(int i = 0; i < n; i++){
            if(seq[i] > n){
                mp[seq[i]] = i + 1;
                mxVal = max(mxVal, seq[i]);
            }
        }
        int ans = 1;
        int cur = 0;
        for(int i = mxVal; i > n; i--){
            if(mp.find(i) != mp.end()){
                cur++;
                continue;
            }
            ans = (ans * cur) % MOD;
        }
        return (ans * n) % MOD;
    }
    for(int i = 0; i < n; i++){
        if(seq[i] <= n){
            int idx = 0;
            for(int j = 0; j < n; j++){
                if(seq2[j + seq[i] - 1] == seq[i]){
                    idx = j;
                    break;
                }
            }
            map<int, int> mp;
            int mxVal = 0;
            for(int i = idx; i < idx + n; i++){
                if(seq2[i] > n){
                    mp[seq2[i]] = i - idx + 1;
                    mxVal = max(mxVal, seq2[i]);
                }
            }
            int ans = 1;
            int cur = 0;
            for(int i = mxVal; i > n; i--){
                if(mp.find(i) != mp.end()){
                    cur++;
                    continue;
                }
                ans = (ans * cur) % MOD;
            }
            return ans;
        }
    }
    return 0;
}
// int main(){
//     int n;
//     cin >> n;
//     int gondolaSeq[n];
//     for(int i = 0; i < n; i++){
//         cin >> gondolaSeq[i];
//     }
//     cout << countReplacement(n, gondolaSeq) << "\n";
// }
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…