提供一种解决方法。

在累加统计每一个国家获得的陨石时，可能会写下这样的代码。

for(auto pos:v[a[i]])cnt+=tr.ask(pos);

实际上，我们只关心当前收集的陨石有没有达到要求，因为 $p_i$ 很小，可以这样改动。

for(auto pos:v[a[i]]){
    cnt+=tr.ask(pos);
    if(cnt>=p[a[i]])break;
}

这样就可以通过本题啦！