Editorial for WC '16 Contest 1 S4 - TP
Submitting an official solution before solving the problem yourself is a bannable offence.
Computing expected values directly can be problematic, but they can often be computed based on probabilities of events instead. For example, in the case of this problem, if we let  be the probability that exactly 
 different houses will have been TP-ed after all 
 passes, then the expected number of different houses to be TP-ed is 
.
Now, how can we compute these  values? We can use dynamic programming. Let 
 be the probability that exactly 
 houses will have been TP-ed after 
 passes. For starters, we know that 
, and 
. From a given state 
, there are only 
 possible transitions to consider: whether the 
-th pass results in 
, 
, or 
 additional houses getting TP-ed (leading to states 
, 
, or 
). We'll need to compute the probability of each of these 
 transitions occurring. For starters, there are 
 total possible pairs of houses, and if 
 houses have been TP-ed so far, then 
 houses have not.
In order for zero new houses to be TP-ed, both targeted houses must have already been TP-ed. The number of such houses is . Therefore, the probability of this transition is 
. As a result, we add 
 onto 
.
Similarly, there are  pairs of targeted houses which would result in 
 new house getting TP-ed, and there are 
 pairs which result in 
 new houses getting TP-ed. In the same way, we can add 
 onto 
, and 
 onto 
.
Finally, once we've considered all states for  and 
, we'll have populated the whole 
 array. There are 
 states and only a constant number (
) of transitions from each state, meaning that the time complexity of this algorithm is 
. For each 
, 
, so we can proceed to calculate the expected value as described initially.
This problem can also be solved in  with some more advanced math. For any given house, 
 pairs of houses result in it getting TP-ed on any given pass. Therefore, the probability of it getting TP-ed on any given pass is 
, and the probability of it not getting TP-ed is therefore 
. As such, the probability of any given house not getting TP-ed at all across all 
 passes is 
. Due to the property of linear expectation, the expected number of houses which won't get TP-ed at all is simply 
 times that value. Thus, the expected number of houses which will get TP-ed is 
.
Comments