92pts求调
查看原帖
92pts求调
119120
Waddles楼主2024/9/30 12:46

WA了3,4两个点,求大佬指正

#include<bits/stdc++.h>
using namespace std;
char s[1000005];
int n,m,top,kmp[1000005];
int dep[1000005],lg[1000005],fa[1000005][30],head[1000005];
struct Node
{
    int nxt;
    int to;
}a[2000005];
void add(int x,int y)
{
    a[++top].nxt=head[x];
    a[top].to=y;
    head[x]=top;
}
void dfs(int x,int fat)
{
    dep[x]=dep[fat]+1;
    fa[x][0]=fat;
    for(int i=1;i<=lg[dep[x]];i++)
    {
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=head[x];i;i=a[i].nxt)
    {
        int y=a[i].to;
        if(y==fat)continue;
        dfs(y,x);
    }
}
int query(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    while(dep[x]>dep[y])
        x=fa[x][lg[dep[x]-dep[y]]];
    if(x==y)return fa[x][0];
    for(int i=lg[dep[x]];i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];
            y=fa[y][i];
        }
    }
    return fa[x][0];
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    scanf("%d",&m);
    int j=0;
    add(1,0);
    add(0,1);
    for(int i=2;i<=n;i++)
    {
        while(j&&s[i]!=s[j+1])j=kmp[j];
        if(s[i]==s[j+1])j++;
        kmp[i]=j;
        add(i,j);
        add(j,i);
    }
    for(int i=1;i<=n;i++)
    {
        lg[i]=lg[i-1]+((1<<(lg[i-1]+1))==i);
    }
    dfs(0,0);
    while(m--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        printf("%d\n",query(x,y));
    }
    return 0;
}
2024/9/30 12:46
加载中...