比赛 2025.3.29 评测结果 AAAAAAAAAAAAAAAAAAAAAAAAA
题目名称 硝华流焰 最终得分 100
用户昵称 zqy 运行时间 1.172 s
代码语言 C++ 内存使用 17.00 MiB
提交时间 2025-03-29 10:17:26
显示代码纯文本
#include <iostream>
#include <cstdio>
#define int long long
using namespace std;
const int N=1e5+10,mod=1e9+7;
int n,ans,f[N][8],g[N][8],a[N];
int ver[N],to[N<<1],nxt[N<<1],val[N<<1],idx;
void add(int x,int y,int z){
	to[++idx]=y,nxt[idx]=ver[x],ver[x]=idx,val[idx]=z;
} 
void dfs(int x,int fa){
    g[x][1<<a[x]]=1;
    for(int i=ver[x],y,z;i;i=nxt[i]){
    	y=to[i],z=val[i];
        if(y==fa)continue;
        dfs(y,x);
        for(int i=1;i<=7;i++){
            for(int j=1;j<=7;j++){
                if((i|j)==7&&g[x][i]&&g[y][j]){
                    ans=(ans+f[x][i]*g[y][j]%mod+f[y][j]*g[x][i]%mod+z*g[x][i]*g[y][j]%mod)%mod;
                }
            }
        }
        for(int j=1,i;j<=7;j++){
            if(!g[y][j])continue;
            i=((1<<a[x])|j);
            f[x][i]=(f[x][i]+f[y][j]+z*g[y][j])%mod;
            g[x][i]=(g[x][i]+g[y][j])%mod;
        }
    }
    return;
}
signed main(){
	freopen("blossom.in","r",stdin);
	freopen("blossom.out","w",stdout);
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)scanf("%lld",a+i);
    for(int i=1,u,v,w;i<n;i++){
        scanf("%lld %lld %lld",&u,&v,&w);
        add(u,v,w),add(v,u,w);
    }
    dfs(1,0);
    printf("%lld\n",ans);
    return 0;
}