三角果计数

给一个n个节点的树, 三角果定义为一个包含3个节点的集合, 且他们两两之间的最短路长度a, b, c能够构成一个三角形。
做法：
和边权没什么关系，画画图发现一条链上所有点都不符合条件，得从这条链上有一个分支，才可以构成合法的。
枚举每个点作为下面这样子的中心节点：这里2是中心节点。

     1
     |
     2
    / \
   3   4

接下来是计数，假设先选上中心的上面的某个点，还有两个点在下面选；然后是三个点都在下面选。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 1e6, mod = 998244353;
vector<int> v[N];
int sum[N];
int ans;
int n;
void dfs(int x, int fa)
{
    sum[x] = 1;
    for (auto i : v[x])
    {
        int y = i;
        if (y == fa) continue;
        dfs(y, x);
        sum[x] += sum[y];
    }
    int num = sum[x] - 1;
    for (auto i : v[x]) 
    {
        int y = i;
        if (y == fa) continue;
        ans += sum[y] * (num - sum[y]) * (n - sum[x]);
        num -= sum[y];
    }


    num = sum[x] - 1;
    int sum1 = 0;
    for (auto i : v[x]) 
    {
        int y = i;
        if (y == fa) continue;
        ans += sum1 * sum[y] * (num - sum1 - sum[y]);
        sum1 += sum[y];
    }
}
signed main ()
{
    cin >> n;
    for (int i = 1; i <= n - 1; i++){
        int x, y, z;
        scanf("%lld%lld%lld", &x, &y, &z);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1, 0);
    cout << ans;
    return 0;
}