| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 520720 | peti1234 | Split the Attractions (IOI19_split) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
int n, m, db[4];
vector<int> valasz, el[100000];
void dfs(int a, int ert) {
if (!db[ert]) return;
valasz[a]=ert;
db[ert]--;
for (auto x:el[a]) {
if (!valasz[x]) {
dfs(x, ert);
}
}
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
int m=p.size();
db[1]=a, db[2]=b, db[3]=c;
for (int i=0; i<m; i++) {
int x=p[i], y=q[i];
el[x].push_back(y), el[y].push_back(x);
}
valasz.resize(n);
// 2
if (a==1) {
dfs(0, 2);
for (int i=0; i<n; i++) {
if (!valasz[i]) {
if (db[1]) {
valasz[i]=1;
db[1]--;
} else {
valasz[i]=3;
db[3]--;
}
}
}
return valasz;
}
// 1
if (m<n) {
// lanc
for (int i=0; i<n; i++) {
if (el[i].size()==1) {
if (db[1]) {
dfs(i, 1);
} else {
dfs(i, 2);
}
}
}
} else {
// kor
dfs(0, 1);
for (int i=0; i<n; i++) {
if (!valasz[i] && db[2]) {
dfs(i, 2);
}
}
}
for (int i=0; i<n; i++) {
if (!valasz[i]) {
valasz[i]=3;
db[3]--;
}
}
return valasz;
}
int main() {
return 0;
}
/*
int b1, b2, b3, b4, b5;
vector<int> b6, b7, b8;
int main()
{
cin >> b1 >> b2 >> b3 >> b4 >> b5;
for (int i=1; i<=b2; i++) {int x; cin >> x; b6.push_back(x);}
for (int i=1; i<=b2; i++) {int x; cin >> x; b7.push_back(x);}
b8=find_split(b1, b3, b4, b5, b6, b7);
for (int i=0; i<b8.size(); i++) cout << b8[i] << " ";
return 0;
}
*/
/*
9 10 3 3 3
0 0 0 0 0 0 1 2 4 5
1 3 4 6 7 8 2 3 5 6
*/