CCF 202012-4 食材运输

预处理出cost[i][j]表示第i 个食材从第j个点出发运送完所需的时间，t[i][S]表示从i点出发送完S 集合中出现的食材所需的最大时间，f[i][S]表示选了i个点，已送完了S种食材所需的最小时间，最后答案就是min{f[i][(1<<k)-1]} (i<=m)

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define LL long long
using namespace std;
const int N=2e3;
struct node{
	int y,n,v;
}e[N];
LL lin[N],deep[110],need[N][N],cost[N][N],t[110][2000],f[110][2000],v[N],len=0,ans,minn,maxn,n,m,k,x,y,val;
void LK(int x,int y,int val){
	e[++len].y=y;
	e[len].n=lin[x];
	e[len].v=val;
	lin[x]=len;
}
void dfs1(int k,int x,int pre){
	if(need[x][k]==1)maxn=max(maxn,deep[x]);
		
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==pre||v[y]==0)continue;
		deep[y]=deep[x]+e[i].v;
		ans+=e[i].v*2;
		dfs1(k,y,x);
	}
}
void dfs2(int k,int x,int pre){
	v[x]=need[x][k];
	for(int i=lin[x];i;i=e[i].n){
		int y=e[i].y;
		if(y==pre)continue;
		dfs2(k,y,x);
		v[x]|=v[y];
	}
}
int main(){
	scanf("%d%d%d",&n,&m,&k);

	rep(i,1,n)rep(j,1,k)scanf("%d",&need[i][j]);
	rep(i,2,n){
		scanf("%d%d%d",&x,&y,&val);
		LK(x,y,val),LK(y,x,val);
	}
	rep(i,1,k){
		rep(j,1,n){
			ans=0;
			maxn=0;
			memset(v,0,sizeof(v));
			memset(deep,0,sizeof(deep));
			dfs2(i,j,j);
			dfs1(i,j,j);
			cost[i][j]=ans-maxn;
		}
	}
	rep(i,1,n){
		for(int S=1;S<(1<<k);S++){
			LL ans=0;
			rep(j,1,k){
				if(S&(1<<(j-1))){
					ans=max(ans,cost[j][i]);
				}
			}
			t[i][S]=ans;
		}
	}
	memset(f,0x3f,sizeof(f));
	f[0][0]=0;
	rep(i,1,n){
		rep(j,1,m){
			rep(S,1,(1<<k)-1){
				for(int U=S;U;U=(U-1)&S){
					
					if(f[j-1][S^U]<1e15)f[j][S]=min(f[j][S],max(f[j-1][S^U],t[i][U]));
				}
			}
		}
	}
	ans=1e15;
	rep(i,1,m)ans=min(ans,f[i][(1<<k)-1]);
	cout<<ans;
	return 0;
}