小 Y 的问题

大水题一道，合法个数用 乘法原理+组合数公式
具体做法：num[i]表示与i相连的点的个数
我们称图上的B-D边为中心边
我们枚举中心边
枚举两个点，先看一侧的点i，它的邻接点有num[i]个，但是其中一个点是中心边的另一个点，符合条件的点有num[i]-1个，利用组合数公式可以求出在这些点中选2个的方案有x种，那么Y总数就要加上x*(num[另一个点]-1),另一个同理。
至于最大值，我们预处理每个点连接的边的权值前三大
为啥是前三大呢？因为前两大中可能有有一条中心边，所以要维护前三大值，在一侧找出符合条件的边的前两大值，在另一侧找出最大值算出总和就行了
注意Y总数要开long long型

#include <iostream>
#include <cstdio>
using namespace std;
struct tw{
    int maxn;
    int d;
}maxm[200010][4];
long long num[200010];
int x[200010],y[200010];
int len[200010];
int maxf;
long long cnt;
void max_add(int a,int z,int b)
{
    int s1=maxm[a][1].maxn,s2=maxm[a][2].maxn,s3=maxm[a][3].maxn;
    int d1=maxm[a][1].d,d2=maxm[a][2].d,d3=maxm[a][3].d;    
    if(z>=s1)
    {
    maxm[a][1].maxn=z,maxm[a][1].d=b;
    maxm[a][2].maxn=s1,maxm[a][2].d=d1;
    maxm[a][3].maxn=s2,maxm[a][3].d=d2;
    return;
    }    
    if(z>=s2)
    {
    maxm[a][2].maxn=z,maxm[a][2].d=b;
    maxm[a][3].maxn=s2,maxm[a][3].d=d2;
    return;
    }   
    if(z>=s3)
    {
    maxm[a][3].maxn=z,maxm[a][3].d=b;
    return;
    }       

}
long long cnm(long long x)
{
    return (x-1)*(x)/2;
}
int max11,max12,max2;
bool f[4];
void find_max1(int a,int d)
{
    f[1]=0,f[2]=0,f[3]=0;
    if(maxm[a][1].d!=d)
     max11=maxm[a][1].maxn,f[1]=1;
     else
      max11=maxm[a][2].maxn,f[2]=1;
    if(maxm[a][2].d!=d&&f[2]==0)
      max12=maxm[a][2].maxn;
       else
        max12=maxm[a][3].maxn;
}
int find_max2(int a,int d)
{
    if(maxm[a][1].d!=d)
     return maxm[a][1].maxn;
    return maxm[a][2].maxn;
}
int main()
{
    //freopen("question.in","r",stdin);
    //freopen("question.out","w",stdout);
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++)
     {
        int a,b,z;
        scanf("%d%d%d",&a,&b,&z);
        x[i]=a,y[i]=b,num[a]++,num[b]++,len[i]=z;
        max_add(a,z,b),max_add(b,z,a);
     }
    for(int i=1;i<=n-1;i++)
     {
        int a=x[i],b=y[i],lenx=len[i];
        if(num[a]>=3&&num[b])
        {
            cnt+=cnm(num[a]-1)*(num[b]-1);
            max11=0,max12=0,max2=0;
            find_max1(a,b),max2=find_max2(b,a);
            maxf=max(maxf,max11+max12+max2+lenx);
        }
        if(num[b]>=3&&num[a])
        {
            cnt+=cnm(num[b]-1)*(num[a]-1);
            max11=0,max12=0,max2=0;
            find_max1(b,a),max2=find_max2(a,b);
            maxf=max(maxf,max11+max12+max2+lenx);
        }
     }

     printf("%lld\n%d",cnt,maxf);

     return 0;
}