首页 > 程序开发 > 综合编程 > 其他综合 >

bzoj2286 虚树+树形dp

2017-03-17

bzoj2286 虚树+树形dp:给定一棵根节点为1的带边权的树。每次给定一些点,求把这些点与树根断开的最小花费。

bzoj2286 虚树+树形dp:给定一棵根节点为1的带边权的树。每次给定一些点,求把这些点与树根断开的最小花费。

题目分析:
我们可以O(n)处理出每个点与根断开的最小花费,即该节点到根的路径上的最小边权。然后我们对于每一个询问,可以把询问的点打上标记,然后可以O(n)动态规划求解。

但是O(n)的时间复杂度接受不了,而且我们发现每次询问并不会询问所有的点,而且询问的总点数不超过50w。
这样我们可以每次只把这些询问的点拎出来,建一棵虚树,在虚树上dp。这样时间复杂度上界为O(总点数*2)
这道题就可以过了。

代码如下:

#include 
#include 
#include 
#define N 250020
using namespace std;
inline int Min(int x,int y) { return x&#39;9&#39; ||  ch<&#39;0&#39;) ch=getchar();
    while(ch<=&#39;9&#39; && ch>=&#39;0&#39;)
    {
        c=(c<<1)+(c<<3)+ch-&#39;0&#39;;
        ch=getchar();
    }
    return;
}
int fir[N],nes[N<<1],v[N<<1],q[N<<1],tot=1;
int cost[N],a[N],xl[N],sta[N];
int fa[N][19],dep[N],pos[N],top;
int n,m,K;
bool b[N];
long long f[N],ans;
void edge(int x,int y,int z=0)
{   
    tot++;
    v[tot]=y;
    q[tot]=z;
    nes[tot]=fir[x];
    fir[x]=tot;
    return;
}
#define Edge_Round(x,y,z) edge(x,y,z),edge(y,x,z)
void dfs(int c)
{
    pos[c]=++top;
    dep[c]=dep[fa[c][0]]+1;
    for(int i=1;i<19;i++)
        fa[c][i]=fa[fa[c][i-1]][i-1];
    for(int t=fir[c];t;t=nes[t])
    {
        if(v[t]==fa[c][0]) continue;
        fa[v[t]][0]=c;
        cost[v[t]]=Min(cost[c],q[t]);
        dfs(v[t]);
    }
}
int lca(int x,int y)
{
    if(dep[x]=dep[y]) x=fa[x][i];
    if(x==y) return x;
    for(int i=18;~i;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
bool cmp(int a,int b) { return pos[a]0)
        {
            if(top>1 && dep[sta[top-1]]>dep[LCA]) edge(sta[top-1],sta[top]),top--;
            else if(dep[sta[top]]>dep[LCA]) { edge(LCA,sta[top]),top--; break;}
            else break;
        }
        if(LCA!=sta[top]) sta[++top]=LCA;
        sta[++top]=a[i];
    }
    while(top>1) edge(sta[top-1],sta[top]),top--;
    top=0;
    dfs2(1);
    printf("%lld\n",f[1]);
    for(int i=1;i<=top;i++)
    {
        b[xl[i]]=false;
        fir[xl[i]]=0;
    }
    return;

}
int main()
{
    read(n);
    for(int i=1,x,y,z;i
        
   
相关文章
最新文章
热点推荐