2021牛客暑期多校训练营8 J-Tree（思维dp+rmq）

J-Tree

Cosmic_Tree题解

赛时队友想到了预处理$s\to t$路径上的点最远到达的点，后面贪心做的一直wa。正解是dp。

---

下面的$\text{solve}(l,r,\text{op})$函数中，当$l=r$时不满足题意不合法，当$\text{op}=0$表示当前是第一个人操作，也就是第二个人走到了不合法的$l=r$的局面，而第二个人想要结果尽可能小，因此需要返回$\text{Inf}$，类似的$\text{op}=1$时返回$-\text{Inf}$

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
template <class T=int> T rd()
{
    T res=0;T fg=1;
    char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') fg=-1;ch=getchar();}
    while( isdigit(ch)) res=(res<<1)+(res<<3)+(ch^48),ch=getchar();
    return res*fg;
}
const int N=500010;
int h[N],e[2*N],ne[2*N],idx;
void add(int a,int b){e[idx]=b,ne[idx]=h[a],h[a]=idx++;}
int n,s,t;
int d[N],to[N];
int f[N],cnt;
int dfs(int u,int fa)
{
    int fl=0;
    if(u==t) fl=1;
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int v=e[i];
        if(v==fa) continue;
        int tmp=dfs(v,u);
        fl|=tmp;
        if(tmp) 
            to[u]=v;
        else
            d[u]=max(d[u],d[v]+1);
    }
    return fl;
}
int L[N][21],R[N][21];
int lg[N];
void init()
{
    lg[0]=1;
    for(int i=2;i<=cnt;i++) lg[i]=lg[i>>1]+1;
    for(int i=1;i<=cnt;i++) L[i][0]=f[i]+i-1;
    for(int i=1;i<=cnt;i++) R[i][0]=f[i]+cnt-i;
    
    for(int j=1;j<=lg[cnt];j++)
        for(int i=1;i+(1<<j)-1<=cnt;i++)
        {
            L[i][j]=max(L[i][j-1],L[i+(1<<j-1)][j-1]);
            R[i][j]=max(R[i][j-1],R[i+(1<<j-1)][j-1]);
        }
}
int ql(int l,int r){int len=lg[r-l+1];return max(L[l][len],L[r-(1<<len)+1][len]);}
int qr(int l,int r){int len=lg[r-l+1];return max(R[l][len],R[r-(1<<len)+1][len]);}
int solve(int l,int r,int op)
{
    if(l==r)
    {
        if(op==0) 
            return 0x3f3f3f3f;
        else
            return -0x3f3f3f3f;
    }
    if(op==0) 
        return max((f[l]+l-1)-qr(l+1,r),solve(l+1,r,1));
    else
        return min(ql(l,r-1)-(f[r]+cnt-r),solve(l,r-1,0));
}
int main()
{
    memset(h,-1,sizeof h);
    n=rd(),s=rd(),t=rd();
    for(int i=1;i<n;i++)
    {
        int u=rd(),v=rd();
        add(u,v),add(v,u);
    }
    dfs(s,0);
    f[++cnt]=d[s];
    int tmp=s;
    while(tmp!=t) {tmp=to[tmp];f[++cnt]=d[tmp];}
    
    init();
    printf("%d\n",solve(1,cnt,0));
    return 0;
}