成长,生命,幸福 problem 7452
对于一个度为K的节点,成长一次后会生成2个度为2的节点以及K-2个度为3 的节点,成长两次后会生成2*2+2*(K-2)个度为2的点以及K-2个度为3的点……
设成长一次后会生成X个度为2的节点以及Y个度为3 的节点(X+Y=K)
X1=2,X2=2*X1+2*Y,X3=2*X2+2*Y……
化简,Xm=2^(m-1)*(X1+2Y)-2Y =(2^m-1)*(K-1)+1
套用求树的直径的板子,调用两次dfs,把每次找到一个子节点加的深度1改为加上该节点的度数,同时调用上式对当前最大直径进行更新(K替换为当前直径,1替换为当前直径上经过的节点数)
当m较小时,直接带入公式计算即可;m较大时,令(2^m-1)=1e9带入计算
code
const int mod = 1e9 + 7;
const int N = 100000 + 10;
int M = 1e9;
int n, c, m;
int d[N];
int cnt = 0, cntt = 0;
vector<int> E[N],ind(N);
int qsm(int a, int b){
int ans = 1;
if (a == 0) return 0;
a %= mod;
for (; b; b >>= 1) {
if (b & 1)
ans = 1ll * ans * a % mod;
a = 1ll * a * a % mod;
}
return ans % mod;
}
void dfs(int u, int fa,int num) {
for (int v : E[u]) {
if (v == fa) continue;
d[v] = d[u] + ind[u];
if ((d[v] - num - 1ll) * M + num + 1ll > (d[c]-cntt) * M + cntt) {
c = v;
cntt = num+1ll;
}
dfs(v, u, num + 1);
}
}
void solve() {
cin >> n >> m;
c = 0;
if (m <= 30)
M = qsm(2, m) - 1;
else
M = 1e9;
for (int i = 0; i <= n; i++) {
E[i].clear();
ind[i] = 0;
d[i] = 0;
}
for (int i = 1; i < n; i++) {
int u, v;
cin >> u >> v;
E[u].push_back(v), E[v].push_back(u);
ind[v]++;
ind[u]++;
}
dfs(1, 0, 1);
cntt= 0;
d[c] = 0, dfs(c, 0, 1);
d[c]++;
int ans = 0;
ans = (((d[c] - cntt) % mod * (qsm(2, m) - 1 + mod) % mod) % mod + cntt + mod) % mod;
cout << (ans + mod) % mod << '\n';
}