51nod 1610 路径计数

因为是无环图，所以可以用记忆化搜索做。

可以根据公约数，将图拆成100个子图。

由于有重复，所以再加上容斥。

#include<bits/stdc++.h>
using namespace std;

const long long mod=1e9+7;
const int MAXM=50050;
long long dp[110];
int a[MAXM],b[MAXM],c[MAXM];
int mat[110][110][110];
long long f[110],g[110];
vector<int> vec;
int n;

long long dfs(int u,int d)
{
	if(dp[u]!=-1)
		return dp[u];
	long long ret=0;
	for(int i=1;i<=n;i++)
	{
		if(mat[d][u][i])
			ret=(ret+mat[d][u][i]*(1+dfs(i,d)))%mod;
	}
	return dp[u]=ret;
}

long long cal(int u)
{
	memset(dp,-1,sizeof(dp));
	long long ret=0;
	for(int i=1;i<=n;i++)
		ret=(ret+dfs(i,u))%mod;
	return ret;
}

int main()
{
	int m,i,j,T,x,y;
	while(~scanf("%d%d",&n,&m))
	{
		memset(mat,0,sizeof(mat));
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a[i],&b[i],&c[i]);
			for(j=1;j*j<=c[i];j++)
			{
				if(c[i]%j==0)
				{
					mat[j][a[i]][b[i]]++;
					if(c[i]/j!=j)
						mat[c[i]/j][a[i]][b[i]]++;
				}
			}
		}
		for(i=1;i<=100;i++)
			g[i]=cal(i);
		for(i=100;i>0;i--)
		{
			f[i]=g[i];
			for(j=2*i;j<=100;j+=i)
				f[i]=(f[i]-f[j]+mod)%mod;
		} 
		printf("%lld\n",f[1]);
		scanf("%d",&T);
		while(T--)
		{
			scanf("%d%d",&x,&y);
			vec.clear();
			for(i=1;i*i<=c[x];i++)
			{
				if(c[x]%i==0)
				{
					mat[i][a[x]][b[x]]--;
					vec.push_back(i);
					if(c[x]/i!=i)
					{
						mat[c[x]/i][a[x]][b[x]]--;
						vec.push_back(c[x]/i);
					}	
				}
			}
			c[x]=y;
			for(i=1;i*i<=c[x];i++)
			{
				if(c[x]%i==0)
				{
					mat[i][a[x]][b[x]]++;
					vec.push_back(i);
					if(c[x]/i!=i)
					{
						mat[c[x]/i][a[x]][b[x]]++;
						vec.push_back(c[x]/i); 
					}
				}
			}
			for(i=0;i<vec.size();i++)
				g[vec[i]]=cal(vec[i]);
			for(i=100;i>0;i--)
			{
				f[i]=g[i];
				for(j=i+i;j<=100;j+=i)
					f[i]=(f[i]-f[j]+mod)%mod;
			}
			printf("%lld\n",f[1]);
		}
	}
}