USACO 2019 December Contest, Platinum
Problem 2. Bessie's Snow Cow
Contest has ended.
Log in to allow submissions in analysis mode
Bessie has added a nose to one of the snowballs, so it represents the head of the abstract snow cow. She designates it as snowball number 1. To add more visual interest, she plans to dye some of the snowballs different colors in an artistic fashion by filling old milk pails with colored dye and splashing them onto the sculpture. Colors are identified by integers in the range $1 \ldots 10^5$, and Bessie has an unlimited supply of buckets filled with dyes of every possible color.
When Bessie splashes a snowball with a bucket of dye, all the snowballs in its subtree are also splashed with the same dye (snowball $y$ is in the subtree of snowball $x$ if $x$ lies on the path from $y$ to the head snowball). By splashing each color with great care, Bessie makes sure that all colors a snowball has been splashed with will remain visible. For example, if a snowball had colors $[1,2,3]$ and Bessie splashes it with color $4$, the snowball will then have colors $[1,2,3,4]$.
After splashing the snowballs some number of times, Bessie may also want to know how colorful a part of her snow-cow is. The "colorfulness" of a snowball $x$ is equal to the number of distinct colors $c$ such that snowball $x$ is colored $c$. If Bessie asks you about snowball $x$, you should reply with the sum of the colorfulness values of all snowballs in the subtree of $x.$
Please help Bessie find the colorfulness of her snow-cow at certain points in time.
SCORING:
$Q$ is defined below.
- Test cases 2-3 satisfy $N\le 10^2, Q\le 2\cdot 10^2.$
- Test cases 4-6 satisfy $N\le 10^3, Q\le 2\cdot 10^3.$
INPUT FORMAT (file snowcow.in):
The first line contains $N,$ and the number of queries $Q$ ($1\le Q\le 10^5$).The next $N-1$ lines each contain two space-separated integers $a$ and $b,$ describing a branch connecting snowballs $a$ and $b$ ($1 \le a, b \le N$).
Finally, the last $Q$ lines each contain a query. A query of the form
1 x c
indicates that Bessie splashed a bucket of juice of color $c$ on snowball $x,$ coloring all snowballs in the subtree of $x$. A line of the form
2 x
is a query for the sum of the colorfulness values of all snowballs in the subtree of $x$. Of course, $1\le x\le N$ and $1\le c\le 10^5.$
OUTPUT FORMAT (file snowcow.out):
For each query of type 2, print the sum of colorfulness values within the corresponding subtree. Note that you should use 64-bit integers to avoid overflow.SAMPLE INPUT:
5 18 1 2 1 3 3 4 3 5 1 4 1 2 1 2 2 2 3 2 4 2 5 1 5 1 2 1 2 2 2 3 2 4 2 5 1 1 1 2 1 2 2 2 3 2 4 2 5
SAMPLE OUTPUT:
1 0 1 1 0 2 0 2 1 1 5 1 3 1 1
After the first query of type 1, snowball 4 is dyed with color 1.
After the second query of type 1, snowballs 4 and 5 are dyed with color 1.
After the third query of type 1, all snowballs are dyed with color 1.
Problem credits: Michael Cao and Benjamin Qi