hdu4916 Count on the path,树dp

6 篇文章 0 订阅
水题搞了半天也是醉了。
以1号节点为根,如果路径不经过根就是1.如果经过根,拆为两条,找这两颗子树中不在路径的最小值和另外一些子树的最小值。
大概过程就是dfs(),找每个点往下的最小值。dfs3(),找每个点从上往下不在路径上的最小值。
dfs2()找每个点所在的子树。

很丑陋的代码,但是明天要比赛,懒得改了!

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define NN 1001000
#define inf 1000000100

struct edge{int v,ne;}ed[NN*2];
int te,fi[NN];

int n;

int branch[NN],smin[10],sb[10],f[NN];
int ch[NN][2],son[NN];

void init_edge(){
    te=-1;
    memset(fi,-1,sizeof(fi));
}

void addedge(int fr,int to){
    te++;ed[te].ne=fi[fr];fi[fr]=te;ed[te].v=to;
}

void dfs(int u,int fa){
    int e,v,tmp;
    f[u]=inf;
    ch[u][0]=ch[u][1]=inf;
    son[u]=-1;
    for(e=fi[u];e!=-1;e=ed[e].ne){
        v=ed[e].v;
        if (v==fa) continue;
        dfs(v,u);
        tmp=min(v,f[v]);
        if (tmp<ch[u][0]){
            ch[u][1]=ch[u][0];ch[u][0]=tmp;son[u]=v;
        }
        else if (tmp<ch[u][1]){
            ch[u][1]=tmp;
        }
    }
    f[u]=ch[u][0];
}

void dfs3(int u,int fa,int mi){
    f[u]=min(f[u],mi);
    int e,v;
    for(e=fi[u];e!=-1;e=ed[e].ne){
        v=ed[e].v;
        if (v==fa) continue;
        if (v==son[u]){
            dfs3(v,u,min(mi,ch[u][1]));
        }
        else {
            dfs3(v,u,min(mi,ch[u][0]));
        }
    }
}

void dfs2(int u,int fa,int val){
    int e,v;
    branch[u]=val;
    for(e=fi[u];e!=-1;e=ed[e].ne){
        v=ed[e].v;
        if (v==fa) continue;
        dfs2(v,u,val);
    }
}

int query(int a,int b){
    int i;
    if (a>b) swap(a,b);
    if (a==1&&b==1) return 2;
    else if (a==1){
        for(i=1;i<=3;++i){
            if (branch[b]!=sb[i]) break;
        }
        return min(smin[i],f[b]);
    }
    else {
        if (branch[a]==branch[b]) return 1;
        for(i=1;i<=3;++i){
            if (!(branch[a]==sb[i]||branch[b]==sb[i])) break;
        }
        return min(f[a],min(smin[i],f[b]));
    }
}

int main(){
    int q;
    int i,a,b,e,v,ans,la,tmp;
    while(scanf("%d%d",&n,&q)!=EOF){
        init_edge();
        for(i=1;i<n;++i){
            scanf("%d%d",&a,&b);
            addedge(a,b);
            addedge(b,a);
        }
        memset(f,1,sizeof(f));
        dfs(1,-1);
        for(e=fi[1];e!=-1;e=ed[e].ne){
            v=ed[e].v;
            dfs3(v,1,inf);
        }
        f[1]=inf;
        branch[1]=1;
        for(e=fi[1];e!=-1;e=ed[e].ne){
            dfs2(ed[e].v,1,ed[e].v);
        }
        for(i=1;i<=5;++i){
            smin[i]=inf;sb[i]=-1;
        }
        for(e=fi[1];e!=-1;e=ed[e].ne){
            v=ed[e].v;
            tmp=min(v,f[v]);
            if (tmp<smin[1]){
                smin[3]=smin[2];smin[2]=smin[1];smin[1]=tmp;
                sb[3]=sb[2];sb[2]=sb[1];sb[1]=v;
            }
            else if (tmp<smin[2]){
                smin[3]=smin[2];smin[2]=tmp;
                sb[3]=sb[2];sb[2]=v;
            }
            else if (tmp<smin[3]){
                smin[3]=tmp;
                sb[3]=v;
            }
        }
        la=0;
        for(i=1;i<=q;++i){
            scanf("%d%d",&a,&b);
            a=a^la;b=b^la;
            ans=query(a,b);
            printf("%d\n",ans);
            la=ans;
        }
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值