bzoj1316(点分治)

感觉自己点分治还是太慢了。。

这个可以弄成m次点分治做，不过合在一起做效果更佳。。。

/**
 *　　　　　　  　　┏┓　　 　┏┓
 * 　　　　　  　　┏┛┗━━━━━━━┛┗━━━┓
 * 　　　　　  　　┃　　　　　　　┃ 　
 * 　　　　　  　　┃　　　━　　 　┃
 * 　　　　　  　　┃　＞　　　＜　┃
 * 　　　　　  　　┃　　　　　　　┃
 * 　　　　　  　　┃...　⌒　... 　┃
 * 　　　　　　  　┃              ┃
 * 　　　　　　  　┗━┓          ┏━┛
 * 　　　　　　　　　┃          ┃　Code is far away from bug with the animal protecting　　　　　　　　　　
 * 　　　　　　　　　┃          ┃   神兽保佑,代码无bug
 * 　　　　　　　　　┃          ┃　　　　　　　　　　　
 * 　　　　　　　　　┃          ┃  　　　　　　
 * 　　　　　　　　　┃          ┃
 * 　　　　　　　　　┃          ┃　　　　　　　　　　　
 * 　　　　　　　　　┃          ┗━━━┓
 * 　　　　　　　　　┃              ┣┓
 * 　　　　　　　　　┃              ┏┛
 * 　　　　　　　　　┗┓┓┏━━━━━━━━┳┓┏┛
 * 　　　　　　　　　　┃┫┫       ┃┫┫
 * 　　　　　　　　　　┗┻┛       ┗┻┛
 */ 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#include<map>
#include<stack>
#include<set>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define eps 1e-12
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define mid (x+y>>1)
#define NM 10005
#define nm 20005
#define N 1000005
#define pi 3.1415926535897931
const int inf=1000000007;
using namespace std;
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}


struct edge{int t,v;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y,int v){o->t=y;o->v=v;o->next=h[x];h[x]=o++;}
int n,m,d[NM],size[NM],smin,root,b[NM],_x,_y,_t,tot,cnt,c[NM];
bool v[NM],_v[N],ans[NM];

void dfs1(int x,int f){size[x]=1;link(x)if(!v[j->t]&&j->t!=f)dfs1(j->t,x),size[x]+=size[j->t];}
void getroot(int x,int f){
    int s=tot-size[x];
    link(x)if(!v[j->t]&&j->t!=f)getroot(j->t,x),s=max(s,size[j->t]);
    if(s<smin)smin=s,root=x;
}

void dfs2(int x,int f){
    if(d[x]>1e6)return;
    b[++cnt]=x;
    inc(i,1,m)if(d[x]<=c[i]&&_v[c[i]-d[x]])ans[i]=true;
    link(x)if(!v[j->t]&&j->t!=f){
	d[j->t]=d[x]+j->v;
	dfs2(j->t,x);
    }
}

void div(int x){
    dfs1(x,0);
    tot=size[x];smin=inf;
    getroot(x,0);
    v[root]++;cnt=0;
    link(root)if(!v[j->t]){
	int t=cnt;
	d[j->t]=j->v;
	dfs2(j->t,root);
	while(++t<=cnt)_v[d[b[t]]]=true;
    }
    inc(i,1,cnt)_v[d[b[i]]]=false;_v[0]=true;
    link(root)if(!v[j->t])div(j->t);
}

int main(){
    n=read();m=read();
    inc(i,1,n-1){_x=read();_y=read();_t=read();add(_x,_y,_t);add(_y,_x,_t);}
    inc(i,1,m)c[i]=read();
    inc(i,1,m)if(!c[i])ans[i]=true;
    _v[0]++;
    div(1);
    inc(i,1,m)if(ans[i])puts("Yes");else puts("No");
    return 0;
}

1316: 树上的询问

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1204 Solved: 348
[ Submit][ Status][ Discuss]

Description

一棵n个点的带权有根树，有p个询问，每次询问树中是否存在一条长度为Len的路径，如果是，输出Yes否输出No.

Input

第一行两个整数n, p分别表示点的个数和询问的个数．接下来n-1行每行三个数x, y, c，表示有一条树边x→y，长度为c．接下来p行每行一个数Len，表示询问树中是否存在一条长度为Len的路径．

Output

输出有p行，Yes或No.

Sample Input

6 4
1 2 5
1 3 7
1 4 1
3 5 2
3 6 3
1
8
13
14

Sample Output

Yes
Yes
No
Yes

HINT

30%的数据，n≤100．
100%的数据，n≤10000，p≤100，长度≤1000000．

做完此题可看下POJ 3237 Tree

Source

[ Submit][ Status][ Discuss]