2018牛客网暑期ACM多校训练第三场 G-Coloring Tree

题目链接：G-Coloring Tree

 

题意：N个节点的树，可以染色K种

要求相同颜色节点的距离为D，求问方案总数。

 

分析：BFS  

构造函数f(D) 表示相同颜色节点距离至少为D的方案总数，那么最后答案就是f(D)-f(D+1)

对于当前节点，找出已经访问过的节点中与当前节点的距离小于D 的节点。

这些节点任意两个之间的距离一定小于 D ,所以这些节点的颜色是互不相同的。

记录总个数为cnt ，则当前节点可以染的颜色种数为k−cnt 。

答案就累乘一下，时间复杂度 O(n2)

 

代码如下：

#include 
#include 
#include 
#include 
#include 
using namespace std;

const int mod = 1000000007;
const int maxn = 5000+10;
int n,k,d;
vectorg[maxn];
queueque;
int vis[maxn];
int ta[maxn];
int col;

inline int add(int a, int b) {
   a+=b;
   return a>=mod?a-=mod:a;
}

inline int sub(int a, int b){
    a-=b;
    return a<0?a+=mod:a;
}

inline int mul(long long a, int b){
    a*=b;
    return a>=mod?a%mod:a;
}

int dfs(int u, int fa, int dep){
    if (dep<=0) return 1;
    int ret = 1;
    for (int i=0; i= k) return 0;
        res = mul(res,k-cnt);

        for (int i=0; i