제출 #879445

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
879445		JakobZorz	곤돌라 (IOI14_gondola)	C++14	80 / 100	32 ms	5316 KiB

gondola

#include"gondola.h"
#include<iostream>
#include<unordered_set>
#include<vector>
#include<algorithm>
using namespace std;

int valid(int n,int inputSeq[]){
    unordered_set<int>s(inputSeq,inputSeq+n);
    if((int)s.size()!=n){
        return 0;
    }
    
    int curr=-1;
    for(int i=0;i<2*n;i++){
        if(curr==-1&&inputSeq[i%n]<=n)
            curr=inputSeq[i%n];
        
        if(inputSeq[i%n]<=n&&curr!=inputSeq[i%n]){
            return 0;
        }
        
        if(curr!=-1){
            curr++;
            if(curr==n+1)
                curr=1;
        }
    }
    return 1;
}

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

int replacement(int n,int gondolaSeq[],int replacementSeq[]){
    int len=0;
    int mx=0;
    for(int i=0;i<n;i++)
        mx=max(mx,gondolaSeq[i]);
    len=mx-n;
    for(int i=0;i<len;i++)
        replacementSeq[i]=0;
    
    bool present=false;
    for(int i=0;i<n;i++)
        if(gondolaSeq[i]<=n)
            present=true;
    
    if(len==0)
        return 0;
    
    
    if(present){
        int curr=-1;
        for(int i=0;i<2*n;i++){
            if(curr==-1&&gondolaSeq[i%n]<=n)
                curr=gondolaSeq[i%n];
            
            if(i>=n&&gondolaSeq[i%n]<mx&&gondolaSeq[i%n]>n){
                replacementSeq[gondolaSeq[i%n]-n-1]=curr;
            }
            
            curr++;
            if(curr==n+1)
                curr=1;
        }
        
        curr=-1;
        for(int i=0;i<2*n;i++){
            if(curr==-1&&gondolaSeq[i%n]<=n)
                curr=gondolaSeq[i%n];
            
            if(i>=n&&gondolaSeq[i%n]==mx){
                for(int i=0;i<len;i++){
                    if(replacementSeq[i]==0){
                        replacementSeq[i]=curr;
                        curr=i+n+1;
                    }
                }
                break;
            }
            
            curr++;
            if(curr==n+1)
                curr=1;
        }
    }else{
        for(int i=0;i<n;i++){
            if(gondolaSeq[i]<mx&&gondolaSeq[i]>n){
                replacementSeq[gondolaSeq[i]-n-1]=i+1;
            }
        }
        for(int i=0;i<n;i++){
            if(gondolaSeq[i]==mx){
                int curr=i+1;
                for(int i=0;i<len;i++){
                    if(replacementSeq[i]==0){
                        replacementSeq[i]=curr;
                        curr=i+n+1;
                    }
                }
            }
        }
    }
    
    return len;
}

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

const int MOD=1e9+9;
typedef long long ll;

ll powm(ll base,ll exp){
    ll res=1;
    for(int i=0;i<=30;i++){
        if(exp&(1<<i)){
            res=res*base%MOD;
        }
        base=base*base%MOD;
    }
    return res;
}

int countReplacement(int n,int inputSeq[]){
    vector<int>arr;
    if(valid(n,inputSeq)==0)
        return 0;
    
    for(int i=0;i<n;i++)
        if(inputSeq[i]>n)
            arr.push_back(inputSeq[i]);
    sort(arr.begin(),arr.end());
    ll res=1;
    int prev=n;
    for(int i=0;i<(int)arr.size();i++){
        res=res*powm(arr.size()-i,arr[i]-prev-1)%MOD;
        prev=arr[i];
    }
    
    bool present=false;
    for(int i=0;i<n;i++)
        if(inputSeq[i]<=n)
            present=true;
    
    if(!present)
        res=res*n%MOD;
    
    return(int)res;
}

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 10 / 10
• Subtask 4: 5 / 5
• Subtask 5: 10 / 10
• Subtask 6: 0 / 20
• Subtask 7: 5 / 5
• Subtask 8: 15 / 15
• Subtask 9: 15 / 15
• Subtask 10: 10 / 10