[51nod1588] 幸运树 [树上计数]

[$\mathfrak{L}\mathfrak{i}\mathfrak{n}\mathfrak{k}$]
原题CF109C

---

求树上所有的三元组(i,j,k) i≠j≠k 满足i到j, k的路径中至少有一条边只含有4或者7。
1≤n≤10^5, 1≤wi≤10^9

---

i是题目提到的两条路径的公共端点。
可以考虑从i入手，那么这道题就是一道树上计数问题。
那么，考虑点i我们可以把树分成三个部分：i为根的子树内（不包括i）外、还有i。
那么分别记为d_i, u_i有f_i = d_i(d_i-1) + u_I(u_i-1) + 2d_iu_i
这也是套路了。
第一、二个就是j≠k的j,k数，其中第一个代表两个都在子树内、第二个代表都在子树外
然后第三个就是子树内外组合一下。

---

上面那个是直接计数的部分，还要考虑需要转移的部分（d_i,u_I）
就可以细化到边来推啦。考虑合并上当前边之后这两个会怎么变化。
假如这个边满足条件，
d_father += size_son
u_son = size_root - size_son
否则，
d_father += d_son
u_son = u_father + d_father - d_son
这些式子也还是都是满满的套路啦。

转移可以bfs/dfs。我比较懒就只写dfs啦。

---

另外还可以搞容斥+连通块。

---

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<ctime>
using namespace std;
#define add_edge(a,b,c) nxt[++tot]=head[a], head[a]=tot, to[tot]=b, val[tot]=c
int head[100005];
int val[200005];
int nxt[200005];
int to[200005];
int u[100005];
int d[100005];
long long f[100005];
int s[100005];
int n, tot;
long long ans;
bool check(int x) {
	int t;
	while (x) {
		t = x % 10;
		if ((t!=4)&&(t!=7)) return 0;
		x /= 10;
	}
	return 1;
}
void getsize(int x, int fa) {
	s[x] = 1;
	for (int i = head[x]; i; i = nxt[i]) {
		if (to[i] == fa) continue;
		getsize(to[i], x);
		s[x] += s[to[i]];
	}
}
void dfs(int x, int fa) {
	for (int i = head[x]; i; i = nxt[i]) {
		if (to[i] == fa) continue;
		dfs(to[i], x);
		if (check(val[i])) {
			d[x] += s[to[i]];
			u[to[i]] = s[1] - s[to[i]];
		} else {
			d[x] += d[to[i]];
//			u[to[i]] = u[x] + d[x] - d[to[i]];
		}
	}
}
void ddfs(int x, int fa) {
	for (int i = head[x]; i; i = nxt[i]) {
		if (to[i] == fa) continue;
		if (check(val[i])) {
//			d[x] += s[to[i]];
//			u[to[i]] = s[1] - s[to[i]];
		} else {
//			d[x] += d[to[i]];
			u[to[i]] = u[x] + d[x] - d[to[i]];
		}
		ddfs(to[i], x);
	}
}
int main() {
	scanf("%d", &n);
	for (int uu, v, w, i = 1; i < n; ++i) {
		scanf("%d%d%d", &uu, &v, &w);
		add_edge(uu, v, w);
		add_edge(v, uu, w);
	}
	getsize(1, 0);
	dfs(1, 0);
	ddfs(1, 0);
	for (int x = 1; x <= n; ++x) {
		f[x] = 1ll * d[x] * (d[x] - 1) + 1ll * u[x] * (u[x] - 1) + 2ll * d[x] * u[x];
		ans += 1ll * f[x];
	}
	printf("%lld", ans);
	return 0; 
}