hdu4916 Count on the path,树dp

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

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

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include
#include
#include
#include
#include
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 (tmpb) 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


你可能感兴趣的:(dp,HDU)