- 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;
}