2019 计蒜之道 初赛 第一场- C. 商汤AI园区的n个路口（中等）

•  33.54%
•  1000ms
•  262144K

北京市商汤科技开发有限公司建立了新的 AI 人工智能产业园，这个产业园区里有 nn 个路口，由 n - 1n−1 条道路连通。第 ii 条道路连接路口 u_iui​ 和 v_ivi​。

每个路口都布有一台信号发射器，信号频段是 11 到 mm 之间的一个整数。

道路所连接的两个路口的发射信号叠加可能会影响道路的正常运行。具体地，如果第 ii 条道路连接的两个路口发射信号的频段分别为 aa 和 bb，那么 \gcd(a, b)gcd(a,b) 不能恰好等于道路的保留频段 w_iwi​。每条道路的保留频段是唯一的，即不会与其余任何道路的保留频段相同。

你现在需要确定每个路口发射信号的频段，使其符合要求。

在开始之前，你想先算出共有多少种合法的方案。

由于答案可能很大，输出对 10 ^ 9 + 7109+7 取模的值作为答案。

输入格式

第一行，两个正整数 n, mn,m 分别代表路口数量和信号频段上限。

接下来 n - 1n−1 行，每行描述一条道路。第 ii 行有三个整数 u_i, v_i, w_iui​,vi​,wi​ 意义如上所述。

保证 1 \le n \le m, 1 \le u_i, v_i \le n, 1 \le w_i \le m1≤n≤m,1≤ui​,vi​≤n,1≤wi​≤m。

输出格式

输出一个整数，代表合法方案的数量对 10 ^ 9 + 7109+7 取模的值。

数据范围

• m \le 10 ^ 3m≤103

样例解释

所有合法的方案为 (2, 2, 1), (2, 2, 3), (3, 3, 1), (3, 3, 2), (3, 3, 3)(2,2,1),(2,2,3),(3,3,1),(3,3,2),(3,3,3)。

样例输入复制

3 3
1 2 1
1 3 2

样例输出复制

5

地址：https://nanti.jisuanke.com/t/39262

思路：树状DP.

对于样例来说，从叶子节点考虑，其实是叶子节点到根节点的所有情况的乘积之和，即从2->1的情况和3->1的情况的乘积之和。

而2->1时，节点1的w={1,2,3}的方案数为{0,1,1},而 3->1时，节点1的w={1,2,3}的方案数为{3,2,3}，因此ans=0*1+1*2+1*3=5

那么就可以用树状DP来求解，而对于__gcd(i,j)是否为w，则可以预处理出 i的__gcd(i,j)=w的所有j

Code:

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
typedef pair<int,int> pr;

const int MAX_N=1e3+5;
const int MAX_M=1e3+5;
const LL MOD=1e9+7;
int n,m,T;
vector<pr> e[MAX_N];
vector<int> d[MAX_N][MAX_M];
LL dp[MAX_N][MAX_M];

void DFS(int u,int pre);
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	int u,v,w;
	for(int i=1;i<n;++i)
	{
		cin>>u>>v>>w;
		e[u].push_back({v,w});
		e[v].push_back({u,w});
	}
	for(int i=1;i<=m;++i)
		for(int j=1;j<=m;++j)
			d[i][__gcd(i,j)].push_back(j);
	DFS(1,0);
	cout<<dp[1][0]<<endl;

	return 0; 
}

void DFS(int u,int pre)
{
	for(int i=1;i<=m;++i)
		dp[u][i]=1;
	int v,w;
	LL S1,dd;
	for(auto c:e[u])
		if(c.first!=pre){
			v=c.first;	w=c.second;
			DFS(v,u);
			for(int i=1;i<=m;++i)
			{
				S1=0;
				for(auto x:d[i][w])
					S1=(S1+dp[v][x])%MOD;
				dd=(dp[v][0]-S1+MOD)%MOD;
				dp[u][i]=dp[u][i]*dd%MOD;
			}
		}
	for(int i=1;i<=m;++i)
		dp[u][0]=(dp[u][0]+dp[u][i])%MOD;
}