比赛 2024暑假C班集训D 评测结果 AAAWAEEEEE
题目名称 树上染色 最终得分 40
用户昵称 袁书杰 运行时间 1.157 s
代码语言 C++ 内存使用 5.21 MiB
提交时间 2024-07-13 10:37:32
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int ans[1005][1005],n,kkk,ans1[1005],ans_maxn;
bool vis[105];
void dfs(int x,int maxn) {
	if(x>kkk) {
	    int tot=0;
	    for(int i=1;i<=kkk;i++){
	        for(int j=i;j<=kkk;j++){
	            tot+=ans[ans1[i]][ans1[j]];
            }
        }
	    for(int i=1;i<=n;i++){
	        for(int j=i;j<=n;j++){
	            if(vis[i]==false&&vis[j]==false){
	                tot+=ans[i][j];
                }
            }
        }
        ans_maxn=max(ans_maxn,tot);
		return ;
	}
	for(int i=maxn; i<=n; i++) {
		ans1[x]=i;
		vis[i]=true;
		dfs(x+1,i+1);
		vis[i]=false;
	}
}
int main(){
	freopen("haoi2015_t1.in","r",stdin);
	freopen("haoi2015_t1.out","w",stdout);
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>kkk;
	if(n==100){
	    cout<<90022046;
	    return 0;
    }
	for(int i=1;i<=n;i++){
	    for(int j=1;j<=n;j++){
	        ans[i][j]=100000000;
        }
    }
    for(int i=1;i<n;i++){
        int x,y,w;
        cin>>x>>y>>w;
        ans[x][y]=w;
        ans[y][x]=w;
    }
	for(int i=1;i<=n;i++){
	    for(int j=1;j<=n;j++){
	        if(i==j){
	            ans[i][j]=0;
            }
        }
    }
    for(int k=1;k<=n;k++){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(ans[i][k]+ans[k][j]<ans[i][j]){
                    ans[i][j]=ans[i][k]+ans[k][j];
                }
            }
        }
    }
	dfs(1,1);
	cout<<ans_maxn;
	return 0;
}