Codeforces 538 E. Demiurges Play Again

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include

using namespace std;
#define rep(i,n) for(int i=0; i b，上帝可以是B中大于b的k2-1个数都大于a
// 则取a，排名k1 + (k2 - 1)
if(sta == 0)
{
res += (son - 1);
}
// 若是后手
// 比如子树A取了a,在A中排k1,A中一共有cnt1个,B取了b,在B中排k2,B中一共有cnt2个
// 假设 cnt1 - a < cnt2 - b，上帝可以使B中的元素都大于a
// 则取a,排名为 cnt1 - a + cnt2
else
{
mi = min(mi,cnt[v[e]] - son);
res += cnt[v[e]];
}
}
if(sta) res -= mi;
return res;
}
int main()
{
int a,b;
scanf("%d",&n);
tol = 0;
mst(fst,-1);
for(int i=0; i