联合权值

Posted 2022-11-20 66dzb

【TimeGate】

https://www.luogu.org/problem/P1351

【解题思路】

枚举每一个点；

然后枚举可以连到他的点；

然后对着些点直接统计答案就好了；

#include<cstdio>
#include<iostream>
using namespace std;
const int N=2e5+5,mo=10007;
struct csint to,nxt;a[N*2];
int head[N],ll,v[N];
int n,ans,x,y,maxans;
void init(int x,int y)
    a[++ll].to=y;
    a[ll].nxt=head[x];
    head[x]=ll;

void work(int x)
    int sum=0,ma=0,m=0;
    for(int k=head[x];k;k=a[k].nxt)
        if(v[a[k].to]>ma)m=ma;    ma=v[a[k].to];else
        if(v[a[k].to]>m)m=v[a[k].to];
        ans=(ans+sum*v[a[k].to])%mo;
        sum=(sum+v[a[k].to])%mo;
    
    maxans=max(maxans,ma*m);

int main()

    scanf("%d",&n);
    for(int i=1;i<n;i++)
        scanf("%d%d",&x,&y);
        init(x,y); init(y,x);
    
    for(int i=1;i<=n;i++)scanf("%d",&v[i]);
    for(int i=1;i<=n;i++)work(i);
    printf("%d %d",maxans,(ans*2)%mo);