点分治练习poj1741；poj2114；bzoj2599

在某次模拟赛里(我已经不想提及pku了TAT)，赛后想切了poj2114，然后因为移动左右指针的位置写的特别挫，TLE了。找了好久错误。好像未果来着。并且怀疑了一晚上的人生。我之前写的点分治都尼玛是翔！

今天又研究了一下，结果一眼发现，特么 移动左右指针的时候我的做法可能r移到l左边很多很多才挺(或许根本没有停。。诶。那句话怎么说来着，退一步，海阔天空。

然后今天写了几个。

poj1741

求dis[a]+dis[b]<=k的点对个数n<=10000 k<=1e8

（其实今天并没有写这个= =

只是发现以前的板子是错的。

while(deep[l]+deep[r]>k&&l<r) r--;

纠正一下嘛

#include<iostream>#include<cstdio> #include<algorithm>using namespace std;const int N=10010;const int inf=0x7fffffff;struct E {int to,nxt,val;}edge[N*2];int idx[N],can[N],size[N],deep[N],d[N],f[N];int tot,root,ans,ret,sum,p;int n,k;void init(){    tot=1;root=0;ans=0;sum=n;    fill(idx,idx+N,0);    fill(can,can+N,1);    fill(size,size+N,0);    fill(deep,deep+N,0);    fill(d,d+N,0);    fill(f,f+N,0);f[0]=inf;}void addedge(int from,int to,int val){    edge[tot].to=to;edge[tot].val=val;edge[tot].nxt=idx[from];idx[from]=tot++;}void getroot(int x,int fa){    size[x]=1;f[x]=0;    for(int t=idx[x];t;t=edge[t].nxt){        E e=edge[t];        if(can[e.to] && e.to!=fa){            getroot(e.to,x);            f[x]=max(f[x],size[e.to]);            size[x]+=size[e.to];        }    }    f[x]=max(f[x],sum-size[x]);    if(f[x]<f[root]) root=x;}void getdeep(int x,int fa){    deep[++p]=d[x];    for(int t=idx[x];t;t=edge[t].nxt){        E e=edge[t];        if(can[e.to] && e.to!=fa) {            d[e.to]=d[x]+e.val;            getdeep(e.to,x);        }    }}int calc(int x,int now){    ret=0;d[x]=now;p=0;    getdeep(x,0);    sort(deep+1,deep+1+p);    int r=p;    for(int l=1;l<=p;l++){        while(deep[l]+deep[r]>k&&l<r) r--;        if(r<=l) break;        ret+=(r-l);    }    return ret;}void work(int x){    ans+=calc(x,0);    can[x]=0;    for(int t=idx[x];t;t=edge[t].nxt){        E e=edge[t];        if(can[e.to]){            sum=size[e.to];            ans-=calc(e.to,e.val);            root=0;            getroot(e.to,0);            work(root);        }    }}int main(){    freopen("in.txt","r",stdin);    freopen("out.txt","w",stdout);    while(1){        scanf("%d%d",&n,&k);        if(n==0 && k==0) break;        init();        for(int i=1;i<n;i++){            int x,y,w;            scanf("%d%d%d",&x,&y,&w);            addedge(x,y,w);addedge(y,x,w);        }        getroot(1,0);        work(root);        printf("%d\n",ans);    }    return 0;} 

poj2114

求dis[a]+dis[b]==k的点对是否存在。n<=10000 k<=1e8

一样的。当时是指针没处理好。这次换了hzwer的方法。。其实是通用方法辣

然后WA了一次。当两个指针指的值相加正好等于k时，需要讨论是右指针左移还是左指针右移。

#include<cstdio>#include<cstring>#include<algorithm>#define N 10005#define inf 1<<30using namespace std;struct E{int to,v,nxt;}edge[N*2];int n,k,ans;int idx[N],tot;int sum,root,sz[N],f[N],d[N],dep[N];bool vis[N];void addedge(int from,int to,int v){edge[tot].to=to;edge[tot].v=v;edge[tot].nxt=idx[from];idx[from]=tot++;}void getroot(int x,int fa){sz[x]=1;f[x]=0;for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(e.to==fa||vis[e.to]) continue;getroot(e.to,x);sz[x]+=sz[e.to];f[x]=max(f[x],sz[e.to]);}f[x]=max(f[x],sum-sz[x]);if(f[x]<f[root]) root=x;}void getdep(int x,int fa){dep[++dep[0]]=d[x];for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(e.to==fa||vis[e.to]) continue;d[e.to]=d[x]+e.v;getdep(e.to,x);}}int calc(int x,int now){int ret=0;d[x]=now;dep[0]=0;getdep(x,0);sort(dep+1,dep+1+dep[0]);for(int l=1,r=dep[0];l<r;){if(dep[l]+dep[r]==k){ret++;if(dep[r]==dep[r-1]) r--;else l++;continue;}if(dep[l]+dep[r]<k) l++;else r--;}return ret;}void work(int x){ans+=calc(x,0);vis[x]=1;for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(vis[e.to]) continue;ans-=calc(e.to,e.v);sum=sz[e.to];root=0;getroot(e.to,root);work(root);}}int main(){freopen("in.txt","r",stdin);while(scanf("%d",&n)&&n){tot=1;f[0]=inf;memset(idx,0,sizeof(idx));int x,v;for(int i=1;i<=n;i++)while(scanf("%d",&x)&&x){scanf("%d",&v);addedge(i,x,v);addedge(x,i,v);}while(scanf("%d",&k)&&k){memset(vis,0,sizeof(vis));sum=n;root=0;ans=0;getroot(1,0);work(root);if(ans) puts("AYE");else puts("NAY");}puts(".");}return 0;}

bzoj2599

求dis[a]+dis[b]==k的点对的最小边数。n<=100000 k<=1000000

k辣么小 可以开个数组存辣

对于每次找到的根，跑每个儿子，had数组记录已经跑完的儿子的对应的到根的距离的最小边数。

然后每个儿子跑一边getans and update。。

然后做完一个点再跑一边清零。。

WA了一次。。因为had数组赋inf是居然写的是N。。应该是M。。

#include<cstdio>#include<algorithm>#define inf 1<<30#define N 200005#define M 1000005using namespace std;struct E{int to,v,nxt;}edge[N*2];int idx[N],tot;bool vis[N];int n,k,sum,ans;int root,sz[N],f[N],dis[N],edges[N],had[M];void addedge(int from,int to,int v){edge[++tot].to=to;edge[tot].v=v;edge[tot].nxt=idx[from];idx[from]=tot;}void getroot(int x,int fa){sz[x]=1;f[x]=0;for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(e.to==fa||vis[e.to]) continue;getroot(e.to,x);sz[x]+=sz[e.to];f[x]=max(f[x],sz[e.to]);}f[x]=max(f[x],sum-sz[x]);if(f[x]<f[root]) root=x;}void add(int x,int fa,int flag){if(dis[x]<=k){if(flag) had[dis[x]]=min(had[dis[x]],edges[x]);else had[dis[x]]=inf;}for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(e.to==fa||vis[e.to]) continue;add(e.to,x,flag);}}int calc(int x,int fa){if(k>=dis[x]) ans=min(ans,edges[x]+had[k-dis[x]]);for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(e.to==fa||vis[e.to]) continue;dis[e.to]=dis[x]+e.v;edges[e.to]=edges[x]+1;calc(e.to,x);}}void work(int x){vis[x]=1;had[0]=0;for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(vis[e.to]) continue;dis[e.to]=e.v;edges[e.to]=1;calc(e.to,0);add(e.to,0,1);}//printf("%d\n",x);//for(int i=1;i<=k;i++) printf("%d ",had[i]);puts("");for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(vis[e.to]) continue;add(e.to,0,0);}for(int t=idx[x];t;t=edge[t].nxt){E e=edge[t];if(vis[e.to]) continue;sum=sz[e.to];root=0;getroot(e.to,root);work(root);}}int main(){freopen("in.txt","r",stdin);freopen("out.txt","w",stdout);scanf("%d%d",&n,&k);for(int i=1;i<n;i++){int x,y,v;scanf("%d%d%d",&x,&y,&v);x++;y++;addedge(x,y,v);addedge(y,x,v);}fill(had,had+M,inf);sum=n;ans=inf;f[0]=inf;getroot(1,0);work(root);if(ans==inf) printf("-1\n");else printf("%d\n",ans);return 0;}