#505. 三角果计数

#include<bits/stdc++.h>
using namespace std;

#define rep(i,l,r) for(int i=(l);i<=(r);i++)
#define per(i,l,r) for(int i=(l);i>=(r);i--)
#define ll long long
#define mset(s,t) memset(s,t,sizeof(t))
#define mcpy(s,t) memcpy(s,t,sizeof(t))
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define mp make_pair

typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;   
typedef vector<ll> Vll;               
typedef vector<pair<int, int> > vpii;
typedef vector<pair<ll, ll> > vpll;                        

const ll mod = 1e9 + 7;
//const ll mod = 998244353;
const double pi  = acos(-1.0);

inline ll qmi (ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1) ans = ans * a%mod;
		a = a * a %mod;
		b >>= 1;
	}
	return ans;
}
inline int read () {
	int x = 0, f = 0;
	char ch = getchar();
	while (!isdigit(ch)) f |= (ch=='-'),ch= getchar();
	while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
	return f?-x:x;
}
template<typename T> void print(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x >= 10) print(x/10);
	putchar(x % 10 + '0');
}
inline ll sub (ll a, ll b) {
	return ((a - b ) %mod + mod) %mod;
}
inline ll add (ll a, ll b) {
	return (a + b) %mod;
}
inline ll inv (ll a) {
	return qmi(a, mod - 2);
}
#define int long long
vector<int> g[100005];
int ans;
int n;
int len[100005];
void dfs (int u, int fa) {
	for (auto v : g[u]) {
		if (v != fa) {
			dfs(v, u);
			ans += len[u] * len[v] * (n - len[u] - len[v] - 1);
			len[u] += len[v];
		}
	}
	len[u] ++;
}
void solve() {
	cin >> n;
	for (int i = 1; i < n; i++ ){
		int a,b, w;
		cin >> a >> b >> w;
		g[a].pb(b);
		g[b].pb(a);
	}
	dfs(1, -1);
	cout << ans << endl;
}
signed main () {
	// ios::sync_with_stdio(0),cin.tie(0), cout.tie(0);
    int t;
    t =1;
    //cin >> t;
    while (t --) solve();
    return 0;
}

∑ ∑ ∑ a i a j a k \sum\sum\sum{a_i}{a_j}{a_k} aiajak

首先要明白枚举一条线上的点一定是不能构成三角形的 然后我们可以这么想。我们枚举其中一个点u,该树就会成为多个子树 这样就成为了在盒子里选择数情况了 枚举中点情况就变成了

∑ a i a j ∑ a k \sum{a_i}{a_j}\sum{a_k} aiajak

其中aj是已经枚举了点,a_i是现在正在枚举的点,a_k是将要枚举的点 于是放在树上就是 对于(u, v) len[u] * len[v] * (n - len[u] - len[v] - 1) 减1是自身

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值