比赛 2025.3.29 评测结果 RRRRRRRRRRRRRRRRRRRRRRRRR
题目名称 硝华流焰 最终得分 0
用户昵称 李奇文 运行时间 0.543 s
代码语言 C++ 内存使用 3.58 MiB
提交时间 2025-03-29 11:20:04
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=1e5;
int n,a[N],u[N],v[N],w[N];
int main(){
	freopen("blossom.in","r",stdin);
	freopen("blossom.out","W",stdout);
	std::cin>>n;
	for(int i=1;i<=n;i++){
		std::cin>>a[i];
	}
	int sumu=0;
	for(int i=1;i<n;i++){
		std::cin>>u[i]>>v[i]>>w[i];
		sumu+=u[i];
	}
	if(sumu==n-1){
		int tb[3]={};
		for(int i=2;i<=n;i++){
			tb[a[i]]++;
		}
		tb[a[1]]=1;
		std::cout<<tb[0]*tb[1]*tb[2]<<endl;
	}else{
		
	}
	return 0;
}