比赛 |
2025.3.29 |
评测结果 |
AAAAWWWWTTTTWWWWWTTTTTTTT |
题目名称 |
硝华流焰 |
最终得分 |
16 |
用户昵称 |
彭欣越 |
运行时间 |
24.679 s |
代码语言 |
C++ |
内存使用 |
6.79 MiB |
提交时间 |
2025-03-29 11:06:54 |
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010,mod=1e9+7;
int n,s[N],mk[N],flag,cnt;
int head[N],tot;
ll res;
struct node {
int v,w,nxt;
}e[N*2];
void add (int u,int v,int w) {
e[++tot].v=v;
e[tot].w=w;
e[tot].nxt=head[u];
head[u]=tot;
}
void dfs (int u,ll sum,int a,int b,int c) {
//cout << u <<endl;
if (a==1&&b==1&&c==1) res+=sum;
for (int i=head[u];i;i=e[i].nxt) {
int v=e[i].v,w=e[i].w;
if (mk[v]==1) continue;
mk[v]=1;
if (s[v]==0) dfs(v,(sum+w)%mod,1,b,c);
else if (s[v]==1) dfs(v,(sum+w)%mod,a,1,c);
else dfs(v,(sum+w)%mod,a,b,1);
mk[v]=0;
}
return;
}
int main () {
freopen("blossom.in","r",stdin);
freopen("blossom.out","w",stdout);
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
cin >> n;
for (int i=1;i<=n;i++) {
cin >> s[i];
if (s[i]==0) flag=i,cnt++;
}
for (int i=1;i<n;i++) {
int a,b,c;
cin >> a >> b >> c;
add(a,b,c);
add(b,a,c);
}
if (cnt==1) {
mk[flag]=1;
if (s[flag]==0) dfs(flag,0,1,0,0);
else if (s[flag]==1) dfs(flag,0,0,1,0);
else dfs(flag,0,0,0,1);
mk[flag]=0;
}else{
for (int i=1;i<=n;i++) {
mk[i]=1;
if (s[i]==0) dfs(i,0,1,0,0);
else if (s[i]==1) dfs(i,0,0,1,0);
else dfs(i,0,0,0,1);
mk[i]=0;
}
}
cout << res/2 <<endl;
return 0;
}