题意:给定一个树(连通无回路的无向图);然后给你k,表示一共有k种颜色;
并且n个顶点,n-1条边,而且不会有重边;
但是这个题最难理解的就是Ei的定义,我当时也读了半天没太看懂啥意思;
结果是这个意思:
把i颜色的节点连接起来,那么所有使得i颜色的点的可达的边集就是Ei了;
比如第一个案例:
如果我按照上图进行染色的话,那么连接1颜色的有那么就有这三条边属于集合E1;
对于2颜色来说就是:
这个边,所以他们相交的最大边集的个数就是1;
相信看懂了这个应该就好理解了吧;
其实就是求一种颜色方案使得他们各自的同颜色的点使得这些同颜色的点可达的边形成集合之后,那么这些不同颜色的边集的 交集边数最大是多少条边;
感觉有点绕口,但是也是可以理解滴;
所以问题就来了,怎么构造一种颜色方案使得他们的公共边的数目最多呢?
其实如果你的思维足够快,应该会想到:如果要是一条公共边,那么肯定是都走过的边;那么这个边的左右节点数目会不会有什么关系?
这点可能比较难想了,我也是参考了大佬的思想;
结果就是这个边对应的一端对应的子节点总数应该>=k并且n-一端对应的子节点总数>=k;
为什么会是n-一边对应的节点数>=k?
可以这样理解:
这里我默认的是上面是靠近根节点的节点;所以每个节点存他的子节点的个数+本身的一个节点;
因为有n-1条边,所以n-当前节点的数目就是另外的一端的边数目;
因为只有在>=k的情况下,那么这条边肯定是公共边,这样才能使得公共边的数量最大化;现在问题来了?
怎么求节点的所有子节点个数呢?
dfs吧:
其实这个也可以理解,自己动手把思维理清楚,然后模拟递归就可以理解这个递归怎么写的了;
这个我又学到了怎么求一个节点的子节点数目的dfs方法了;
长见识了;
AC代码:
#include<cstdio>
#include<cstring>
#include<vector>
#include<bits/stdc++.h>
using namespace std;
const int maxn = 200010;
vector<int> edge[maxn];
int a[maxn*4];
void dfs(int son, int fa)//在理解这个递归的时候,不用去看fa这个参数,因为无向图遍历的时候为了不往回遍历,所以用这个来标记前面已经遍历过的父节点,其实这里也可以用一个book数组来标记前面已经被遍历了的点(:
{
a[son]=1;
for(int i=0;i<edge[son].size();i++){
if(edge[son][i]==fa)continue;
dfs(edge[son][i],son);
a[son]+=a[edge[son][i]];
}
}
int main()
{
int t, n, k, u, v;
scanf("%d", &t);
while(t--) {
memset(a, 0, sizeof(a));
scanf("%d %d", &n, &k);
for(int i=0;i<=n;i++)edge[i].clear();
for(int i = 0; i < n-1; i++) {//初始化无向图
scanf("%d %d",&u, &v);
edge[u].push_back(v);
edge[v].push_back(u);
}
int ans = 0;
dfs(1, 0);
for(int i = 1; i <= n; i++) {
if(a[i]>=k&&(n-a[i])>=k)ans++;
}
cout<<ans<<endl;
}
return 0;
}