3개의 배열과 트리

월간 향유회 2024. 12. D번 BOJ 33014번
시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 1024 MB71312740.909%

문제

이 문제는 투 스텝 문제입니다.

준혁이는 당신에게 정점 $N$개의 트리 하나를 빌려주었다. 당신은 트리를 보고 감명받아 길이 $N$의 배열 $3$개에 $2N$보다 작거나 같은 음이 아닌 정수로 값을 채워 넣어 트리를 표현하려고 한다. 각 배열은 다음과 같이 그 가치가 정해진다.

  • 첫 번째 배열은 배열의 값의 합이 그 가치가 된다.
  • 두 번째 배열 또한 배열의 값의 합이 그 가치가 된다.
  • 세 번째 배열은 배열의 값을 모두 XOR한 결과에 $N$을 곱한 값이 그 가치가 된다.

$3$가지 배열의 가치의 합이 $2N \cdot \left \lfloor{\log_{2}N}\right \rfloor^2$를 넘어선 안 된다.

긴 시간이 지난 후, 준혁이는 트리를 다시 가져갔고, 당신은 $3$가지 배열 중 하나의 배열을 잃어버렸다. 당신은 잃어버리지 않은 나머지 두 배열을 보고 어떤 트리를 보고 만든 배열인지 알아내 가져간 트리를 다시 복원해야 한다.

입력

당신의 프로그램은 채점 데이터 하나당 총 두 번 실행된다. 당신은 하나의 소스코드에 두 가지 실행 과정을 모두 구현해야 한다.

모든 입력의 첫 줄에는 실행 단계를 나타내는 정수 $T$가 입력된다. $(1 \leq T \leq 2)$

만약 $T$가 $1$이라면 첫 번째 단계를 수행해야 하고, $T$가 $2$라면 두 번째 단계를 수행하면 된다.

첫 번째 단계

입력

둘째 줄에 $N$이 주어진다. $(4 \leq N \leq 5 \cdot 10^4)$

다음 $N-1$개 줄에 $u$번 정점과 $v$번 정점의 연결을 의미하는 트리의 간선 정보 $u, v$가 주어진다. $(1 \leq u, v \leq N)$

출력

길이 $N$의 배열 $3$개를 배열의 순서대로 한 줄에 하나씩 출력한다. 모든 배열의 원소는 $2N$을 넘지 않는 음이 아닌 정수이고 배열 $3$개의 가치의 합은 $2N \cdot \left \lfloor{\log_{2}N}\right \rfloor^2$을 넘겨선 안된다.

두 번째 단계

입력

둘째 줄에 트리의 정점 개수이자 배열의 길이인 $N$이 주어진다. $(4 \leq N \leq 5 \cdot 10^4)$

세 번째와 네 번째 줄에 첫 번째 단계에서 출력한 배열 $3$개중 $2$개가 한 줄에 하나씩 주어진다. 이때 입력되는 배열 $2$개가 각각 몇 번째 배열이었는지는 알 수 없다.

출력

첫 번째 단계에서 입력된 트리의 간선을 복원하여 각 간선이 연결하는 두 정점 $u$와 $v$를 $N-1$줄에 걸쳐 출력한다. 트리의 번호가 다르거나 간선 집합이 다를 경우 다른 트리이다. 간선의 순서나 $u$와 $v$의 순서는 중요하지 않다.

예제 입력 1

1
5
1 2
1 3
3 4
3 5

예제 출력 1

0 1 2 0 0
0 1 0 1 0
3 1 3 2 0

첫 번째 배열의 가치는 $3$, 두 번째 배열의 가치는 $2$, 세 번째 배열의 가치는 $15$이다.

즉 총 가치의 합은 $20$으로 $2 \cdot 5 \cdot 2^2 = 40$을 넘지 않는다.

예제 입력 2

2
5
3 1 3 2 0
0 1 2 0 0

예제 출력 2

1 2
5 3
3 1
3 4

첫 번째 단계에서 출력한 세 번째 배열과 첫 번째 배열이 입력되었다. 간선의 순서와 $u, v$의 순서를 고려하지 않기 때문에 두 번째 단계에서 출력한 트리와 첫 번째 단계에서 입력된 트리의 간선 집합이 같다.

출처

Contest > BOJ User Contest > 월간 향유회 > 월간 향유회 2024. 12. D번

  • 문제를 만든 사람: gggkik, gunwookim
  • 문제를 검수한 사람: lky7674, snrnsidy, swoon, utilforever

채점 및 기타 정보

  • 예제는 채점하지 않는다.