2024杭电多校2

成长，生命，幸福  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';
}