| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 589218 | fuad27 | Gondola (IOI14_gondola) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include<bits/stdc++.h>
using namespace std;
#define prev(i) ((i-1+n)%n)
typedef long long ll;
typedef long double ld;
const ll inf = 1e18;
const ll mod = 1e9+9;
#define pii pair<int,int>
#define pll pair<ll,ll>
#define vi vector<int>
#define vll vector<ll>
#define rep(i, a, b) for(int i = (a);i<(b);i++)
#define repn(i, a, b) for(int i = (a);i>=(b);i--)
#define ss second
#define ff first
#define mkp make_pair
#define pb push_back
#define all(x) (x).begin(), (x).end()
ll binpow(ll a, ll b) {
ll ans = 1;
while(b) {
if(b&1ll)
ans=(ans*a)%mod;
a=(a*a)%mod;
b>>=1ll;
}
return ans%mod;
}
int valid(int n, int a[])
{
int cnt = a[0]-1;
for(int i = 1;i<n-cnt;i++) {
if(a[i] != a[i-1]+1) {
return 0;
}
}
for(int i = n-cnt+1;i<n;i++) {
if(a[i]!=a[i-1]+1)return 0;
}
if(cnt==0)return 1;
if(a[0] != a[n-1]+1)return 0;
return 1;
return -1;
}
int replacement(int n, int gondolaSeq[], int replacementSeq[])
{
int cnt=0;
for(int i = 0;i<n;i++) {
if(gondolaSeq[i]>n)cnt++;
}
if(cnt == 0)return 0;
else if(cnt == n) {
vector<pair<int,int>> res;
res.push_back(mkp(n,n));
for(int i = 0;i<n;i++) {
res.push_back(mkp(gondolaSeq[i], i+1));
}
sort(res.begin(), res.end());
vector<int> ans;
for(int i=1;i<(int)res.size();i++) {
ans.push_back(res[i].second);
for(int j = res[i-1].first;j<res[i].first-1;j++) {
ans.push_back(j+1);
}
}
for(int i=0;i<(int)ans.size();i++) {
replacementSeq[i]=ans[i];
}
return (int)ans.size();
}
else {
vector<pair<int,int>> res;
res.push_back(make_pair(n,n));
for(int i = 0;i<n;i++) {
if(gondolaSeq[i] > n and gondolaSeq[prev(i)]<=n) {
int tmp = gondolaSeq[i];
gondolaSeq[i] = (gondolaSeq[prev(i)]%n)+1;
res.push_back(make_pair(tmp, gondolaSeq[i]));
}
}
for(int i = 0;i<n;i++) {
if(gondolaSeq[i] > n and gondolaSeq[prev(i)]<=n) {
int tmp = gondolaSeq[i];
gondolaSeq[i] = (gondolaSeq[prev(i)]%n)+1;
res.push_back(make_pair(tmp, gondolaSeq[i]));
}
}
sort(res.begin(), res.end());
vector<int> ans;
for(int i=1;i<(int)res.size();i++) {
ans.push_back(res[i].second);
for(int j = res[i-1].first;j<res[i].first-1;j++) {
ans.push_back(j+1);
}
}
for(int i=0;i<(int)ans.size();i++) {
replacementSeq[i]=ans[i];
}
return (int)ans.size();
}
return 0;
}
int countReplacement(int n, int inputSeq[])
{
vll diff;
int gondolaSeq[n];
rep(i, 0, n)gondolaSeq[i]=inputSeq[i];
vector<pair<ll,ll>> res;
res.push_back(make_pair(n,n));
for(int i = 0;i<n;i++) {
if(gondolaSeq[i] > n and gondolaSeq[prev(i)]<=n) {
ll tmp = gondolaSeq[i];
gondolaSeq[i] = (gondolaSeq[prev(i)]%n)+1;
res.push_back(make_pair(tmp, gondolaSeq[i]));
}
}
for(int i = 0;i<n;i++) {
if(gondolaSeq[i] > n and gondolaSeq[prev(i)]<=n) {
ll tmp = gondolaSeq[i];
gondolaSeq[i] = (gondolaSeq[prev(i)]%n)+1;
res.push_back(make_pair(tmp, gondolaSeq[i]));
}
}
sort(inputSeq, inputSeq+n);
ll ans = 1;
diff.pb(n);
rep(i, 0, n) {
if(inputSeq[i] > n)diff.pb(inputSeq[i]);
}
sort(all(diff));
rep(i, 1, diff.size()) {
ll d = diff[i]-diff[i-1]-1;
if(d>0)ans=(ans*binpow(diff.size()-i, d))%mod;
}
if(!valid(n, gondolaSeq))return 0;
return ans;
}