稀有矿井

第一次知道树形dp还能状压

题面挺长的，大概就是选包含五种矿藏的子树使代价最小。

因为一共只有五种，所以把可以五种拿到与没拿到状态压缩一下，然后就是正常树形 dp 了。

话说我刚开始把 inf 赋成了 2e9 害得自己找了好长时间重写一遍才发现/dk

#include<bits/stdc++.h>
using namespace std;
const int inf=1e9;
int tot,head[1005],nxt[2005],to[2005],w[2005];
int n,a[1005],f[1005][55];
void add(int x,int y,int z){
	nxt[tot]=head[x];to[tot]=y;w[tot]=z;head[x]=tot++;
}
void dp(int x,int pre)
{
	int v=1<<(a[x]-1);
	f[x][v]=0;
	for(int i=head[x];i!=-1;i=nxt[i])
	{
		int y=to[i], z=w[i];
		if(y==pre) continue;
		dp(y,x);
		for(int j=0;j<(1<<5);j++)
		{
			for(int k=0; k<(1<<5); k++)
			{
				f[x][j|k]=min(f[x][j|k],f[x][j]+f[y][k]+z);
			}
		}
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--) 
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			head[i]=-1;
		}
		tot=0;
		for(int i=1;i<n;i++)
		{
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			add(a,b,c);
			add(b,a,c);
		}
	for(int i=1;i<=n;i++)
	{
		f[i][0]=0;
		for(int j=1;j<(1<<5);j++)
		f[i][j]=inf;
	}
	dp(1,-1);
	int ans=inf;
	for(int i=1;i<=n;i++) ans=min(ans,f[i][(1<<5)-1]);
	printf("%d\n",ans);
	}
	return 0;
}