| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 297154 | tmwilliamlin168 | 즐거운 행로 (APIO20_fun) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "fun.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> createFunTour(int n, int q) {
vector<int> s(n), d(n), ans;
array<int, 2> c{n+1};
for(int i=0; i<n; ++i) {
s[i]=attractionsBehind(0, i);
if(s[i]>n/2)
c=min(array<int, 2>{s[i], i}, c);
}
vector<int> e;
for(int i=0; i<n; ++i) {
d[i]=hoursRequired(c[1], i);
if(d[i]==1)
e.push_back(i);
}
vector<vector<int>> d2(e.size()-1), f(3);
for(int i=0; i<e.size()-1; ++i)
for(int j=0; j<n; ++j)
d2[i].push_back(hoursRequired(e[i], j));
for(int i=0; i<n; ++i) {
if(i==c[1])
continue;
int j=0;
while(j<e.size()-1&&d2[j][i]>d[i])
++j;
f[j].push_back(i);
}
for(int i=0; i<e.size(); ++i) {
sort(f[i].begin(), f[i].end(), [&](const int &i, const int &j) {
return d[i]<d[j];
});
}
int p[3]={0, 1, 2};
int l=-1;
while(1) {
sort(p, p+3, [&](const int &i, const int &j) {
return f[i].size()>f[j].size();
});
if(f[p[0]].size()<f[p[1]].size()+f[p[2]].size()) {
sort(p, p+3, [&](const int &i, const int &j) {
return d[f[i].back()]>d[f[j].back();
});
int i=0;
if(p[i]==l)
++i;
ans.push_back(f[p[i]].back());
f[p[i]].pop_back();
l=p[i];
} else
break;
}
f[p[1]].insert(f[p[1]].end(), f[p[2]].begin(), f[p[2]].end());
sort(f[p[1]].begin(), f[p[1]].end(), [&](const int &i, const int &j) {
return d[i]<d[j];
});
l=l==p[0];
while(f[p[l]].size()) {
ans.push_back(f[p[l]].back());
f[p[l]].pop_back();
l^=1;
}
ans.push_back(c[1]);
// for(int a : ans)
// cout << a << endl;
return ans;
}