点分治练习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
,can
,size
,deep
,d
,f
;
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
,tot;
int sum,root,sz
,f
,d
,dep
;
bool vis
;
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
,tot;
bool vis
;
int n,k,sum,ans;
int root,sz
,f
,dis
,edges
,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;
}