点分治练习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<<30
using 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 1000005
using 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;
}