小P的Civilization V

题目内容

 

小P最近在玩Civilization V，游戏的地图是一棵树，树的每个节点都可以当作战场，刚开始每个节点的战斗加成为00

现在小P拥有两种人员：

一种是工人，每个工人会有一个起点uu和终点vv，工人可以使u->vu−>v路径上每个节点的战斗加成+1+1

一种是骑兵，每个骑兵也有一个起点uu和终点vv，骑兵可以选择u->vu−>v路径上任意节点战斗

现有mm个工人和qq个骑兵，问每个骑兵作战最多可以拥有多少战斗加成

输入格式

 

第一行三个整数n,m,qn,m,q

接下来一行n-1n−1个数，第ii个数f_ifi​表示第i+1i+1个节点的父亲

接下来mm行，每行两个整数u,vu,v，表示一个工人

接下来qq行，每行两个整数u,vu,v，表示一个骑兵

输出格式

 

qq行，对于每个骑兵，输出他作战最多可以拥有多少战斗加成

样例 1 输入

 

5 5 5
1 1 1 4 
4 5
4 1
4 3
5 3
4 5
1 3
3 5
1 1
5 2
4 2

样例 1 输出

 

3
5
3
5
5

样例 2 输入

 

9 5 5
1 2 2 4 5 3 4 6 
4 9
4 7
1 7
1 2
2 1
9 9
9 9
3 3
7 7
2 2

样例 2 输出

 

1
1
2
2
4

子任务

子任务名	评分方式	时间限制	内存限制	说明	分数
Subtask	求和	2000 ms	512 MB		100

提示

 

Hint

对于15\%15%的数据，保证n,m,qle 1000n,m,q≤1000

对于另外15\%15%的数据，保证f_i=ifi​=i

对于另外10\%10%的数据，保证所有u=vu=v

对于另外10\%10%的数据，保证工人的u=vu=v

对于另外10\%10%的数据，保证骑兵的u=vu=v

对于100\%100%的数据，保证n,m,qle 500,000n,m,q≤500,000且f_ile ifi​≤i

_________________

没啥难的，树上搜索大集合！

可能是代码能力差，我怎么写这么多搜索！

_________________

#include<bits/stdc++.h>
using namespace std;
const int maxn=5e5+10;
struct edge
{
    int u,v,nxt;
}e[maxn<<1];
int head[maxn],js;
void addage(int u,int v)
{
    e[++js].u=u;e[js].v=v;
    e[js].nxt=head[u];head[u]=js;
}
int n,m,q;
int w[maxn],f[maxn][20],dep[maxn];
void dfs(int u,int fa)
{
    f[u][0]=fa;dep[u]=dep[fa]+1;
    for(int i=1;f[u][i-1];++i)f[u][i]=f[f[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(v==fa)continue;
        dfs(v,u);
    }
}
int ff[maxn][20];
void dfs2(int u,int fa)
{
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(v==fa)continue;
        dfs2(v,u);
        w[u]+=w[v];
    }
}
void dfs3(int u,int fa)
{
    ff[u][0]=w[u];
    for(int i=1;f[u][i-1];++i)ff[u][i]=max(ff[u][i-1],ff[f[u][i-1]][i-1]);
    for(int i=head[u];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if(v==fa)continue;
        dfs3(v,u);
    }
}
int lca(int u,int v)
{
    if(dep[u]<dep[v])swap(u,v);
    for(int i=19;i>=0;--i)if(dep[f[u][i]]>=dep[v])u=f[u][i];
    if(u==v)return u;
    for(int i=19;i>=0;--i)
        if(f[u][i]!=f[v][i])
        {
            u=f[u][i];v=f[v][i];
        }
    return f[u][0];
}
int lca_(int u,int v)
{
    int ans=0;
    if(dep[u]<dep[v])swap(u,v);
    for(int i=19;i>=0;--i)if(dep[f[u][i]]>=dep[v])ans=max(ans,ff[u][i]),u=f[u][i];
    if(u==v)return max(ans,w[u]);
    for(int i=19;i>=0;--i)
        if(f[u][i]!=f[v][i])
        {
            ans=max(ans,max(ff[u][i],ff[v][i]));
            u=f[u][i];v=f[v][i];
        }
    ans=max(max(ans,max(w[u],w[v])),w[f[u][0]]);
    return ans;
}
int main()
{
    scanf("%d%d%d",&n,&m,&q);
    for(int v,i=2;i<=n;++i)
    {
        scanf("%d",&v);
        addage(i,v);addage(v,i);
    }
    dfs(1,0);
    for(int u,v,i=1;i<=m;++i)
    {
        scanf("%d%d",&u,&v);
        w[u]++;w[v]++;
        int l=lca(u,v);
        w[l]--;w[f[l][0]]--;
    }
    dfs2(1,0);
    dfs3(1,0);
    for(int u,v,i=1;i<=q;++i)
    {
        scanf("%d%d",&u,&v);
        printf("%d
",lca_(u,v));
    }
    return 0;
}