【BZOJ 3697】采药人的路径 树上点分治

原创 2017年01月03日 10:17:16

自从gay豪讲了后缀自动机以后就不知道在干什么了,今天闲下来搞搞自己的专题,至于动态点分治什么的真的是坑啊。

好了,回到这道题,同样是处理路径问题,想到树分治。首先考虑没有休息站的方法,这里我想到有两种,一种就是dfs是处理出dis数组,排序后用数学神马的随便搞搞,还有一种就是基于dp思想的,定义数组f[i],g[i]分别表示当前子树中路径长度为i的节点数和之前子树中路径为i的节点数,这样就可以保证求出来的路径一定是经过根节点的。但是这道题除了这个限制以外还有休息站,多一维表示这条路径中有没有休息站,因为休息站一定在两条链之间,在dfs的时候就记录一下当前路径的前缀有那些路径和时出现过就好了。


    #include<cstdio>  
    #include<cstring>  
    #include<iostream>  
    #define LL long long  
    #include<cmath>  
    #include<cstdlib>  
    #define maxn 200020  
    using namespace std;  
    int head[maxn],tot,n,ff[maxn],sz[maxn],rt,size,vis[maxn],cnt[maxn];  
    LL f[maxn][2],g[maxn][2],ans;  
    //f[i][0]表示当前子树中和为i且没有休息站的 方案数1是有休息站  
    struct edge{int v,next,w;}e[maxn];   
    void adde(int a,int b,int c){e[tot].v=b,e[tot].next=head[a],e[tot].w=c;head[a]=tot++;}  
      
    void getrt(int u,int fa){  
        ff[u]=0,sz[u]=1;  
        for(int v,i=head[u];i!=-1;i=e[i].next){  
            if(vis[v=e[i].v]||v==fa)continue;  
            getrt(v,u);  
            sz[u]+=sz[v];  
            ff[u]=max(ff[u],sz[v]);  
        }  
        ff[u]=max(ff[u],size-ff[u]);  
        if(ff[u]<ff[rt])rt=u;  
    }  
    int Max;  
    void dfs(int u,int fa,int sum){  
        if(cnt[sum])f[sum][0]++;  
        else f[sum][1]++;  
        cnt[sum]++;  
        int x=abs(sum-n);  
        Max=max(x,Max);  
        for(int v,i=head[u];i!=-1;i=e[i].next){  
            if(vis[v=e[i].v]||v==fa)continue;  
            dfs(v,u,sum+e[i].w);  
        }  
        cnt[sum]--;  
    }  
      
    void calc(int u){  
        g[n][0]=1;int mx=-1e9;  
        for(int v,i=head[u];i!=-1;i=e[i].next){  
            if(vis[v=e[i].v])continue;Max=-n;  
            dfs(v,u,e[i].w+n);  
            mx=max(mx,Max);  
            ans+=(g[n][1]-1)*f[n][1];  
            for(int j=-Max;j<=Max;j++)  
                ans+=f[n-j][0]*g[n+j][0]+f[n-j][0]*g[n+j][1]+f[n-j][1]*g[n+j][0];  
            for(int j=n-Max;j<=n+Max;j++){  
                g[j][0]+=f[j][0],g[j][1]+=f[j][1];  
                f[j][0]=f[j][1]=0;  
            }  
        }  
        for(int i=n-mx;i<=n+mx;i++)g[i][0]=g[i][1]=0;  
    }  
      
    void solve(int u){  
        vis[u]=1;  
        calc(u);  
        for(int v,i=head[u];i!=-1;i=e[i].next){  
            if(vis[v=e[i].v])continue;  
            ff[rt=0]=size=sz[v];  
            getrt(v,u);  
            solve(rt);  
        }  
    }  
      
    int main(){  
        memset(head,-1,sizeof(head));  
        scanf("%d",&n);  
        for(int a,b,c,i=1;i<n;i++){  
            scanf("%d%d%d",&a,&b,&c);  
            if(c==1)adde(a,b,1),adde(b,a,1);  
            else adde(a,b,-1),adde(b,a,-1);  
        }  
        ff[rt=0]=size=n;  
        getrt(1,0);  
        solve(rt);  
        printf("%lld",ans);  
        return 0;  
    }  


版权声明:你喜欢就好

[BZOJ3697]采药人的路径(点分治)

题目描述传送门题解点分治裸题(笑 黄学长的题解看不懂啊。。。 自己yy了一种做法 果然蠢到爆炸将边权看成1和-1,那么合法的路径即为路径权值和为0,并且从中间分开两段和都是0 如果从路径的一个起...
  • Clove_unique
  • Clove_unique
  • 2017年02月15日 23:29
  • 405

[BZOJ3697] 采药人的路径

传送门http://www.lydsy.com/JudgeOnline/problem.php?id=3697题目大意给定一棵树,找一些路径满足,路径上0,1数量相等,并在路径上找到一个点(休息站),...
  • slongle_amazing
  • slongle_amazing
  • 2016年03月07日 09:14
  • 533

【bzoj3697】【采药人的路径】【点分治】

Description 采药人的药田是一个树状结构,每条路径上都种植着同种药材。 采药人以自己对药材独到的见解,对每种药材进行了分类。大致分为两类,一种是阴性的,一种是阳性的。 采药人每天都...
  • sunshinezff
  • sunshinezff
  • 2016年03月18日 11:18
  • 254

【bzoj3784】【树上的路径】【点分治+堆+st表】

Description 给定一个N个结点的树,结点用正整数1..N编号。每条边有一个正整数权值。用d(a,b)表示从结点a到结点b路边上经过边的权值。其中要求a Input ...
  • sunshinezff
  • sunshinezff
  • 2016年07月08日 09:52
  • 780

初识点剖(基于点的树上分治)

有一类问题,是关于树上路径,树上点对,树上XX…之类的,我们可以考虑运用分治算法. Description 给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K ...
  • jokerwyt
  • jokerwyt
  • 2016年08月20日 09:38
  • 470

【点分治】BZOJ 3697:采药人的路径

BZOJ 3697:采药人的路径Description采药人的药田是一个树状结构,每条路径上都种植着同种药材。 采药人以自己对药材独到的见解,对每种药材进行了分类。大致分为两类,一种是阴性的,一种是...
  • Doom_bringer
  • Doom_bringer
  • 2016年02月03日 10:54
  • 317

BZOJ 3697: 采药人的路径 点分治

3697: 采药人的路径 Time Limit: 10 Sec  Memory Limit: 128 MB Submit: 1329  Solved: 453 [Submit][Status][...
  • BlackJack_
  • BlackJack_
  • 2017年12月03日 12:00
  • 65

BZOJ 3697 采药人的路径 点分治

由于不知道两种颜色的和,但是如果将两种颜色分别设为1和-1,那么符合条件的和必为0,以这个树分治即可。 数组必须可以索引负值为了不调试使代码长了好多。。#include #include #in...
  • huanghongxun
  • huanghongxun
  • 2016年04月05日 23:51
  • 313

bzoj 3697: 采药人的路径 点分治

题意采药人的药田是一个树状结构,每条路径上都种植着同种药材。 采药人以自己对药材独到的见解,对每种药材进行了分类。大致分为两类,一种是阴性的,一种是阳性的。 采药人每天都要进行采药活动。他选择的路...
  • qq_33229466
  • qq_33229466
  • 2017年10月10日 10:30
  • 100

bzoj 3697: 采药人的路径 (点分治)

3697: 采药人的路径 Time Limit: 10 Sec  Memory Limit: 128 MB Submit: 1022  Solved: 347 [Submit][Status][...
  • clover_hxy
  • clover_hxy
  • 2017年03月10日 15:03
  • 173
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:【BZOJ 3697】采药人的路径 树上点分治
举报原因:
原因补充:

(最多只允许输入30个字)