codevs3728联合权值（LCA）

3728 联合权值

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 黄金 Gold

题解

 查看运行结果

 

 

题目描述 Description

输入描述 Input Description

输出描述 Output Description

样例输入 Sample Input

样例输出 Sample Output

数据范围及提示 Data Size & Hint

/*
自己打暴力，看了看不用打LCA只处理倍增数组
然后就10分了......
神奇的LCA 估计打了能有60分
只能看峰峰...
*/

#include<iostream>
#include<cstdio>
#include<cstdlib>
#define maxn 200001
#define mod 10007

using namespace std;
int n,m,cnt,ans,num,tot,x,y,z,maxx;
int head[maxn],deep[maxn],dis[maxn];
int f[maxn][];
struct node
{
    int from;
    int to;
    int dis;
    int next;
}e[maxn*];

int init()
{
    int x=,f=;char c=getchar();
    while(c>''||c<''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x*f;
}

void add(int from,int to)
{
    e[++num].from=from;
    e[num].to=to;
    e[num].next=head[from];
    head[from]=num;
}

void get_fa()
{
    for(int j=;j<=;j++)
      for(int i=;i<=n;i++)
        f[i][j]=f[f[i][j-]][j-];
}

void Dfs(int from,int now,int cnt)
{
    f[now][]=from;deep[now]=cnt;
    for(int i=head[now];i;i=e[i].next)
    {
        int v=e[i].to;
        if(v!=from)
          Dfs(now,v,cnt+);
    }
}

int main()
{
    n=init();
    for(int i=;i<n;i++)
    {
        x=init();y=init();
        add(x,y);add(y,x);
    }
    for(int i=;i<=n;i++)
      dis[i]=init();
    Dfs(,,);
    get_fa();
    for(int i=;i<=n;i++)
      for(int j=;j<=n;j++)
        {
            if(j==i) continue;
            {
                if(deep[i]==deep[j]&&f[i][]==f[j][]||deep[i]-deep[j]==&&f[i][]==j||deep[j]-deep[i]==&&f[j][]==i)
                  {
                       tot+=dis[i]*dis[j];
                       maxx=max(maxx,dis[i]*dis[j]);
                  }
            }
        }
    printf("%d %d\n",maxx,tot%mod);
}

/*
峰峰果然是大神 果然裸地LCA枚举60分
开始的思路 求出每两个点之间的距离（其实枚举两个点就T了）
统计为2 的点对 然后统计答案
倍增LCA的话 是O(n+n*n) 后面的n*n是枚举那两个点
T了 60分
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#define maxn 200010
using namespace std;
int n,w[maxn],fa[maxn][],head[maxn],num,dep[maxn],ans,sum;
struct node
{
    int u,v,pre;
}e[maxn*];
int init()
{
    int x=;char s=getchar();
    while(s<''||s>'')s=getchar();
    while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
    return x;
}
void Add(int from,int to)
{
    num++;
    e[num].u=from;
    e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
void Dfs(int now,int from,int c)
{
    fa[now][]=from;
    dep[now]=c;
    for(int i=head[now];i;i=e[i].pre)
      if(e[i].v!=from)
        Dfs(e[i].v,now,c+);
}
void Get_fa()
{
    for(int j=;j<=;j++)
      for(int i=;i<=n;i++)
        fa[i][j]=fa[fa[i][j-]][j-];
}
int Get_a(int a,int t)
{
    for(int i=;i<=t;i++)
      a=fa[a][];
    return a;
}
int LCA(int a,int b)
{
    if(dep[a]<dep[b])swap(a,b);
    a=Get_a(a,dep[a]-dep[b]);
    if(a==b)return a;
    for(int i=;i>=;i--)
      if(fa[a][i]!=fa[b][i])
        {
          a=fa[a][i];
          b=fa[b][i];
        }
    return fa[a][];
}
int main()
{
    //freopen("linkb.in","r",stdin);
    //freopen("linkb.out","w",stdout);
    n=init();
    int u,v;
    for(int i=;i<=n-;i++)
      {
          u=init();v=init();
          Add(u,v);Add(v,u);
      }
    for(int i=;i<=n;i++)
      w[i]=init();
    Dfs(,,);
    Get_fa();
    for(int i=;i<=n;i++)
      for(int j=i+;j<=n;j++)
        {
          int anc=LCA(i,j);
          int deep=dep[i]+dep[j]-*dep[anc];
          if(deep==)
            {
              ans=max(ans,w[i]*w[j]);
              sum=(sum+w[i]*w[j])%;
            }
        }
    printf("%d %d\n",ans,sum*%);
    return ;
}

/*
因为默认的边权值是1 所以距离为2的点也就是从同一个点出发的两个
这样我们不需要n*n的枚举点 我们统计每个点之间相连的
然后两两组合 这里会发现 又跑慢了 70分
假设点x直接相连的点有三个 权值分别是a b c
那么我们两两组合后总权值是ab+bc+ac
会发现这里和之前用线段树求区间两两元素之和是一样的
根据数学公式 (a+b+c)^2-a^2-b^2-c^2=2*(ab+bc+ac)
这样我们就可以很快地求出权值和 至于最大的嘛 那肯定是最大的两个点权之积
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 200010
#define mod 10007
using namespace std;
int n,w[maxn],head[maxn],num,ans,sum;
struct node
{
    int u,v,pre;
}e[maxn*];
int cmp(int x,int y)
{
    return x>y;
}
int init()
{
    int x=;char s=getchar();
    while(s<''||s>'')s=getchar();
    while(s>=''&&s<=''){x=x*+s-'';s=getchar();}
    return x;
}
void Add(int from,int to)
{
    num++;
    e[num].u=from;
    e[num].v=to;
    e[num].pre=head[from];
    head[from]=num;
}
int main()
{
    //freopen("linkb.in","r",stdin);
    //freopen("linkb.out","w",stdout);
    n=init();
    int x,y;
    for(int i=;i<=n-;i++)
      {
          x=init();y=init();
          Add(x,y);Add(y,x);
      }
    for(int i=;i<=n;i++)
      w[i]=init();
    for(int u=;u<=n;u++)
      {
          int p[maxn],l=,tmp=,s=;
        for(int i=head[u];i;i=e[i].pre)
          p[++l]=w[e[i].v];
        sort(p+,p++l,cmp);
        ans=max(ans,p[]*p[]);
        for(int i=;i<=l;i++)
          {
            tmp=(tmp+p[i]*p[i]%mod)%mod;
            s=(s+p[i]%mod)%mod;
          }
        sum=(sum+(s*s-tmp)%mod)%mod;
      }
    printf("%d %d\n",ans,sum);
    return ;
}