全WA,求助
查看原帖
全WA,求助
1166838
china123楼主2024/3/3 15:04
#include<iostream>
#include<cstring>
using namespace std;
typedef long long LL;
const int N =2e5+10;
int h[N],e[N],ne[N],idx;
int dep[N],f[N][20],_lg[N];
LL sum[N],num[N];
int n,m;
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int p)
{
    f[u][0]=p;
    
    dep[u]=dep[p]+1;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j==p) continue;
        
        sum[j]=sum[u]+num[j];
		dfs(j,u);
    }
}

void unit()
{
    for(int i=2;i<=n;i++) _lg[i]=_lg[1>>i]+1;
    
    for(int j=1;j<20;j++)
        for(int i=1;i<=n;i++)
        {
            f[i][j]=f[ f[i][j-1] ][j-1];
        }
            
}
int lca(int x,int y)
{
    if(dep[x]<dep[y]) return lca(y,x);
    int t=dep[x]-dep[y];
    for(int i=_lg[t];i>=0;i--)
        if(t>>i&1) x=f[x][i];
    if(x==y) return x;
    for(int i=_lg[dep[x]];i>=0;i--)
    {
        if(f[x][i]^f[y][i])
        {
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int main()
{
    cin>>n>>m;
    memset(h,-1,sizeof h);
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);
        num[a]++,num[b]++;
        sum[a]++,sum[b]++;
    }
    
    dfs(1,1);
    unit();
    //for(int i=1;i<=n;i++) cout<<sum[i]<<" ";
    //puts(" ");
    while(m--)
    {
        int a,b;
        cin>>a>>b;
        //cout<<lca(a,b)<<endl;
        //if(lca(a,b)==a) cout<<sum[b]-sum[a]+num[a]<<endl;
        //else if(lca(a,b)==b) cout<<sum[a]-sum[b]+num[b]<<endl;
        cout<<sum[a]+sum[b]-2*sum[lca(a,b)]+num[lca(a,b)]<<endl;
     }
     
    return 0;
}


2024/3/3 15:04
加载中...