题意:每个点都可以涂颜色,求每种颜色连接起来的线的交集最长。
题解:由于是树,即可以从外围一层一层的往里面遍历,将点数叠加起来,到最后面遍历每条边,只要那条边的左右两点都是大于k的这个边就是交集。
ans[k] 表示k点的度。
sum[k] 表示遍历到k点的所有点之和。
du[k] 表示存与k点相连的所有点。
q 表示存度只有一的点,方便遍历。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAX = 200010;
struct node{
int l;
int r;
}a[MAX];
int ans[MAX];
int sum[MAX];
vector <int> du[MAX];
queue <int> q;
int main(){
int T,n,k;
scanf("%d",&T);
while(T--){
scanf("%d %d",&n,&k);
for(int i = 1 ; i <= n ; i++){
du[i].clear();
ans[i]=0;//
sum[i]=1;
}
for(int i = 1 ; i < n ; i++){
scanf("%d %d",&a[i].l,&a[i].r);
du[a[i].r].push_back(a[i].l);
du[a[i].l].push_back(a[i].r);
ans[a[i].r]++;
ans[a[i].l]++;
}
for(int i = 1 ; i <= n ; i++){
if(du[i].size()==1){
q.push(i);
ans[i]--;
}
}
while(!q.empty()){
int k = q.front();
q.pop();
for(int i = 0 ; i < du[k].size() ; i++){
if(ans[du[k][i]]!=0){
ans[du[k][i]]--;
// printf("ans[du[%d][%d]]=%d\n",k,i,ans[du[k][i]]);
if(ans[du[k][i]]==1){
q.push(du[k][i]);
ans[du[k][i]]--;
}
sum[du[k][i]] += sum[k];
// printf("sum[%d]=%d sum[%d]=%d\n",du[k][i],sum[du[k][i]],k,sum[k]);
// printf("ans[%d]=%d\n",du[k][i],ans[du[k][i]]);
}
}
}
int m = 0;
for(int i = 1 ; i < n ; i++){
if(sum[a[i].l]>=k&&sum[a[i].r]>=k){
m++;
}
}
printf("%d\n",m);
}
}