나무 타기
면접 대비시간 제한1초메모리 제한1024 MB
루트에서 리프까지 가는 경로 중 정점 i에서 거리 1부터 A_i 이내의 자손으로만 이동하는 경우의 수를 998244353으로 나눈 나머지로 구합니다.
문제
POSTECH 캠퍼스에는 정점이 개인 루트 트리가 있다. 정점은 1번부터 번까지 번호가 붙어 있고, 번 정점의 부모는 번 정점이다. 루트의 값은 0이다.
POSCAT 부원들은 트리 알고리즘을 공부하려고 이 트리에서 나무 타기 놀이를 한다. 놀이는 루트에서 출발해 리프 방향으로 점프를 반복하며 리프 노드에 도달하는 것이다. 리프 노드는 자식이 없는 노드이다. 각 정점에는 강도 가 주어진다. 번 정점에서 점프할 때는 번 정점과의 거리가 1 이상 이하인 정점으로만 뛸 수 있다. 즉, 현재 정점이 번이면 다음 정점은 번을 루트로 하는 서브트리에 속하고, 번과의 거리가 1 이상 이하여야 한다.
두 놀이의 방문 정점 집합이 다르면 서로 다른 놀이로 본다. 트리가 주어질 때 서로 다른 나무 타기 놀이의 개수를 구하라. 답이 클 수 있으므로 으로 나눈 나머지를 출력한다.
입력
첫 번째 줄에 정점의 수 이 주어진다. () 두 번째 줄에는 각 정점의 부모 노드를 나타내는 개의 정수 이 공백으로 구분되어 주어진다. () 주어지는 그래프가 트리 구조임이 보장된다. 세 번째 줄에는 각 정점의 강도를 나타내는 개의 정수 이 공백으로 구분되어 주어진다. ()
출력
서로 다른 나무 타기 놀이의 개수를 으로 나눈 나머지를 출력한다.