# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
290554 | jairRS | Split the Attractions (IOI19_split) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "split.h"
typedef vector<int> vi;
typedef vector<vi> vvi;
using namespace std;
vvi adj;
vi degree;
int ga, gb, gc;
void dfs(int src, int prev = -1)
{
if (gc > 0)
{
degree[src] = 3;
gc--;
}
else if (gb > 0)
{
degree[src] = 2;
gb--;
}
else if (ga > 0)
{
degree[src] = 1;
ga--;
}
for (int a : adj[src])
if (a != prev)
dfs(a, src);
}
vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q)
{
ga = a;
gb = b;
gc = c;
degree = vi(n, 0);
adj = vvi(n);
for (int i = 0; i < n; i++)
{
adj[p[i]].push_back(q[i]);
adj[q[i]].push_back(p[i]);
degree[p[i]]++;
degree[q[i]]++;
}
for (int i = 0; i < n; i++)
{
if (degree[i] == 1)
{
dfs(i);
break;
}
}
return degree;
}