Gym - 101964C Tree （思维+floyd+树的直径）

2018-2019 ACM-ICPC Southeastern European Regional Programming Contest (SEERC 2018)

C题：

题意：给出一棵树，相邻顶点距离为一，找出精确地m个黑色顶点（白色顶点值为0，黑色为1，这是题目里给出的），使这些定点的最大距离最小。

#include <iostream>
using namespace std;
const int MAX = 521;
const int inf = 0x3f3f3f3f;
int bk[MAX];
int dis[MAX][MAX];
int n,m;
void floyd(){//floyd算法模板
	for (int k = 1; k <= n;k++){
		for (int i = 1; i <= n;i++){
			for (int j = 1; j <= n;j++){
				dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
			}
		}
	}
}
int main(){
	cin >> n >> m;
	for (int i = 1; i <= n;i++){
		for (int j = 1; j <= n;j++){
			if(i==j) dis[i][j]=0;//自己到自己为0
			else dis[i][j]=inf;//初始化为无穷大
		}
	}
	for (int i = 1; i <= n;i++) cin >> bk[i];
	for (int i = 1; i < n;i++){
		int x,y;
		cin >> x >> y;
		dis[x][y]=dis[y][x]=1;//两个点之间的距离为1
	}
	floyd();
	int ans=inf;
	for (int i = 1; i <= n;i++){
		for (int j = i; j <= n;j++){
			if(!bk[i]||!bk[j]) continue;//选择黑色的点
			int sum=0;
			for (int k = 1; k <= n;k++){
				if(bk[k]&&dis[i][k]<=dis[i][j]&&dis[k][j]<=dis[i][j]) sum++;//k点，加入i，j两点之间，看是否使最大距离变化
			}
			if(sum>=m) ans=min(ans,dis[i][j]);//如果满足题目要，求最小值，注意是 >=,这个需要自己好好想想，就是有可能找的时候，找更小的会一次去除多个，而不是想要的数目 
		}
	}
	cout << ans << endl;
	return 0;
}