제출 #297245

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
297245		miss_robot	Gondola (IOI14_gondola)	C++14	100 / 100	79 ms	6008 KiB

gondola

#include <bits/stdc++.h>
#include "gondola.h"

#pragma GCC optimize("O3")
#define ll long long
const ll M = 1e9+9;

using namespace std;

int valid(int n, int inputSeq[]){
	set<int> seen;
	for(int i = 0; i < n; i++){
		if(seen.count(inputSeq[i])) return 0;
		seen.insert(inputSeq[i]);
	}
	int pos = -1, st;
	for(int i = 0; i < n; i++) if(inputSeq[i] <= n) pos = i, st = inputSeq[i];
	if(pos == -1) return 1;
	for(int j = 0; j < n; j++, pos++, pos %= n){
		if(inputSeq[pos] > n) continue;
		if(inputSeq[pos] != (st+j-1)%n+1) return 0;
	}
	return 1;
}

int replacement(int n, int gondolaSeq[], int replacementSeq[]){
	int l = 0, last = n, pos = -1, st;
	vector< pair<int, int> > q;
	vector<int> p(n);
	for(int i = 0; i < n; i++) if(gondolaSeq[i] <= n) pos = i, st = gondolaSeq[i];
	if(pos == -1) for(int i = 0; i < n; i++) p[i] = i+1;
	else for(int i = 0; i < n; i++, pos++, pos %= n) p[pos] = (st+i-1)%n+1;
	for(int i = 0; i < n; i++) if(gondolaSeq[i] > n) q.push_back({gondolaSeq[i], i});
	if(q.empty()) return 0;
	sort(q.begin(), q.end());
	for(auto &t : q){
		replacementSeq[l++] = p[t.second];
		last++;
		while(last != t.first) replacementSeq[l++] = last++;
	}
	return l;
}

ll pw(ll x, ll y){
	if(!y) return 1;
	ll t = pw(x, y/2);
	t = (t*t)%M;
	if(y%2) t = (t*x)%M;
	return t;
}

int countReplacement(int n, int inputSeq[]){
	if(!valid(n, inputSeq)) return 0;
	ll s = 1, m = n;
	vector<int> a;
	for(int i = 0; i < n; i++) if(inputSeq[i] > n) a.push_back(inputSeq[i]);
	a.push_back(n);
	sort(a.begin(), a.end());
	for(int i = 0; i < (int)a.size()-1; i++){
		s *= pw(a.size()-i-1, a[i+1]-a[i]-1);
		s %= M;
	}
	for(int i = 0; i < n; i++) if(inputSeq[i] <= n) m = 1;
	s *= m, s %= M;
	return s;
}

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

gondola.cpp: In function 'int valid(int, int*)':
gondola.cpp:16:16: warning: 'st' may be used uninitialized in this function [-Wmaybe-uninitialized]
   16 |  int pos = -1, st;
      |                ^~
gondola.cpp: In function 'int replacement(int, int*, int*)':
gondola.cpp:27:33: warning: 'st' may be used uninitialized in this function [-Wmaybe-uninitialized]
   27 |  int l = 0, last = n, pos = -1, st;
      |                                 ^~

• 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