15PTS求助
查看原帖
15PTS求助
843293
D2019ly楼主2023/2/9 10:34
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=1e5+5;
const int mod=1e9+7;
int n,r;
struct Tree{
	int nxt,to,val;
}edge[N<<1];
int cnt,head[N<<1];
void add(int u,int v,int w)
{
	cnt++;
	edge[cnt].to=v;
	edge[cnt].nxt=head[u];
	edge[cnt].val=w;
	head[u]=cnt; 
}
int num[N][105],f[N][105];
int g[N][2];
void dfs(int u,int fat)
{
	for(int i=1;i<=r;i++)f[u][i]=i,num[u][i]=1;
	
	for(int i=head[u];i;i=edge[i].nxt)
	{
		int v=edge[i].to;
		if(v==fat)continue;
		int w=edge[i].val;
		dfs(v,u);
		int sum=0;
		for(int j=1;j<=r;j++)
		{
			sum+=num[v][j];
			sum%=mod;
			int val=f[v][j];
			if(val<g[v][0])g[v][1]=g[v][0],g[v][0]=val;
			else if(val<g[v][1])g[v][1]=val;
		}
		if(w==2)
		{
			for(int j=1;j<=r;j++)f[u][j]+=f[v][j],num[u][j]=num[u][j]*num[v][j]%mod;
		}
		if(w==0)
		{
			for(int j=1;j<=r;j++)
			{
//				int tmp=INT_MAX,tot=0;
//				for(int l=1;l<=r;l++)
//				{
//					if(j==l)continue;
//					tmp=min(tmp,f[v][l]);
//					tot+=num[v][l];
//					tot%=mod;
//				}
				int tmp=g[v][0];
				if(tmp==f[v][j])tmp=g[v][1];
				f[u][j]+=tmp;
				num[u][j]=num[u][j]*(sum-num[v][j])%mod;
			}
		}
		if(w==1)
		{
			for(int j=1;j<=r;j++)
			{
//				int tmp=INT_MAX,tot=0;
//				for(int l=1;l<=r;l++)
//				{
//					tmp=min(tmp,f[v][l]);
//					tot+=num[v][l];
//					tot%=mod;
//				}
				int tmp=g[v][0];
				int tot=sum;
				f[u][j]+=tmp;
				num[u][j]=num[u][j]*tot%mod;
			}
		}
	}
}
signed main()
{
	ios::sync_with_stdio(false);
	cin>>n>>r;
	memset(g,0x3f,sizeof(g));
	for(int i=1;i<n;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0);
	int tot=0,minn=LONG_LONG_MAX;
	for(int i=1;i<=r;i++)tot+=num[1][i],minn=min(minn,f[1][i]);
	if(!tot)cout<<"0 0\n";
	else cout<<tot<<" "<<minn<<"\n";
	return 0;
}
2023/2/9 10:34
加载中...