http://poj.org/problem?id=2114
把洛谷点分治1复制过来一直WA?
然后把二分改成two-pointer 过了?
不知道之前的二分哪里写错了。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxl=1e4+10;
int n,m,cnt,k,tot,ans;
int ehead[maxl],dis[maxl],aa[maxl];
struct node
{
int val,num;
}a[maxl];
struct ed
{
int to,nxt,l;
}e[maxl<<1];
bool vis[maxl];
struct centertree
{
int n,ans,mini;
int son[maxl];
inline void dfs(int u,int fa)
{
son[u]=1;int v,res=0;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(v==fa || vis[v]) continue;
dfs(v,u);
son[u]+=son[v];
res=max(res,son[v]);
}
res=max(res,n-son[u]);
if(res<mini)
ans=u,mini=res;
}
int getcenter(int x)
{
ans=0;mini=2e9;
dfs(x,0);
return ans;
}
}tree;
inline void add(int u,int v,int l)
{
e[++cnt].to=v;e[cnt].l=l;
e[cnt].nxt=ehead[u];ehead[u]=cnt;
}
inline void prework()
{
for(int i=1;i<=n;i++)
ehead[i]=0,vis[i]=false;
cnt=0;int u,v,l;
for(int i=1;i<=n;i++)
{
while(~scanf("%d",&v) && v)
{
scanf("%d",&l);
add(i,v,l);add(v,i,l);
}
}
}
inline void getdis(int u,int fa)
{
aa[++aa[0]]=dis[u];
int v;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(v==fa || vis[v]) continue;
dis[v]=dis[u]+e[i].l;
getdis(v,u);
}
}
inline int calc(int u,int w)
{
dis[u]=w;aa[0]=0;
getdis(u,0);
sort(aa+1,aa+1+aa[0]);
tot=1;a[1].val=aa[1];a[1].num=1;
for(int i=2;i<=aa[0];i++)
if(aa[i]!=a[tot].val)
a[++tot].val=aa[i],a[tot].num=1;
else
a[tot].num++;
int sum=0,l=1,r=tot;
while(l<=r)
{
if(a[l].val+a[r].val==k)
{
if(l==r)
{
sum+=a[l].num*(a[l].num-1)/2;
return sum;
}
sum+=a[l].num*a[r].num;
l++;r--;
}
else if(a[l].val+a[r].val<k)
l++;
else
r--;
}
return sum;
}
inline void solv(int u)
{
vis[u]=true;
ans+=calc(u,0);
int v;
for(int i=ehead[u];i;i=e[i].nxt)
{
v=e[i].to;
if(vis[v]) continue;
ans-=calc(v,e[i].l);
tree.n=tree.son[v];
int rt=tree.getcenter(v);
solv(rt);
}
}
inline void mainwork()
{
int rt;
while(scanf("%d",&k) && k)
{
for(int j=1;j<=n;j++)
vis[j]=false;
tree.n=n;ans=0;
rt=tree.getcenter(1);
solv(rt);
if(ans>0)
puts("AYE");
else
puts("NAY");
}
puts(".");
}
int main()
{
while(~scanf("%d",&n) && n)
{
prework();
mainwork();
}
return 0;
}