The confusion about the test example lies in the fact that
if you carefully read the problem (after you tried to send a solution
that got a WA :)  you see that you need to sum up not all the values
(a sum of branch types in the cell) in the given distance, but only
values for different branch types.

Although, I agree that the problem is not described very well.