【NOIP 校内模拟】T1 战争（反向并查集）

我日哦 完全忘了前几天才做的星球大战（JSOI2008

这道题还有花椒麻你 告诉你是一棵树 迷惑你是树上算法

正难即反 考虑离线存下点 用并查集维护

先把没有被破坏的点连起来 当两个联通块将要merge的时候 这时总贡献加了他们的权值和之积（根据乘法分配率可得

简直和星球大战一模一样

#include<bits/stdc++.h>
#define ll long long
#define N 1000005
using namespace std;
const int mod=1000000007;
struct Edge
{
	int from,to,next;
}edge[2*N];
int n,m;
int tot,first[N],father[N];
inline void addedge(int x,int y)
{
	tot++;
	edge[tot].from=x;
	edge[tot].next=first[x];
	edge[tot].to=y;
	first[x]=tot;
}
inline int getfather(int x)
{
	if(father[x]==x)	return x;
	father[x]=getfather(father[x]);
	return father[x];
}
vector <int> attack[N];
bool broken[N];
ll w[N],sum,ans[N];
void merge(int x,int y)
{
	int fx=getfather(x),fy=getfather(y);
	if(fx!=fy)
	{
		father[fx]=fy;
		sum=(sum+w[fx]*w[fy]%mod)%mod;
		w[fy]=(w[fx]+w[fy])%mod;
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);cout.tie(NULL);
	cin>>n>>m;
	for(ll i=1;i<=n;i++)	
	{
		father[i]=i;
		w[i]=i;	
	}
	for(int i=2;i<=n;i++)
	{
		int x;
		cin>>x;
		addedge(i,x);
		addedge(x,i);
	}
	for(int i=1;i<=m;i++)
	{
		int k;
		cin>>k;
		for(int j=1;j<=k;j++)
		{
			int x;
			cin>>x;
			attack[i].push_back(x);
			broken[x]=true;
		} 
	}
	for(int u=1;u<=2*m;u++)	//merge现存的边 
	{
		int x=edge[u].from,y=edge[u].to;
		if(broken[x]||broken[y])	continue;
		merge(x,y);
	}
	ans[m+1]=sum;
	for(int i=m;i>=1;i--)
	{
		for(int j=0;j<attack[i].size();j++)
		{
			int now=attack[i][j];
			broken[now]=false;
			for(int u=first[now];u;u=edge[u].next)
			{
				int vis=edge[u].to;
				if(broken[vis])	continue;
				merge(now,vis);
			}
		}
		ans[i]=sum;
	}
	for(int i=1;i<=m+1;i++)	cout<<ans[i]<<'\n';
	return 0;
}