Luogu P3806 點分治模板1
阿新 • • 發佈:2019-02-23
splay pan sof 要求 模板 保存 include namespace bsp
題意:
給定一棵有n個點的樹詢問樹上距離為k的點對是否存在。
分析:
這個題的詢問和點數都不多(但是顯然暴力是不太好過的,即使有人暴力過了)
這題應該怎麽用點分治呢。顯然,一個模板題,我們直接用套路,每次找重心,對於這個重心處理,過當前點的符合要求的路徑。
我們可以看到這個最大長度1e7,開數組是開得下的,所以維護一個數組(或者bitset也行)來保存之前的子樹中到根距離的長度,之後訪問到每個子樹時,先查詢,再記錄,最後刪除標記,將所有的點都統計完之後,答案便被保存到了數組中,直接按照題意輸出即可。
代碼:
1 #include<bits/stdc++.h> 2點分治using namespace std; 3 const int N=100005,inf=1e7; 4 struct node{int y,z,nxt;}e[N<<1]; 5 int n,m,c=0,h[N],mx[N],siz[N],d[N],rem[N],sm; 6 int t[inf],jd[inf],vis[N],que[N],q[N],rt,ans; 7 void add(int x,int y,int z){ 8 e[++c]=(node){y,z,h[x]};h[x]=c; 9 e[++c]=(node){x,z,h[y]};h[y]=c; 10} void getrt(int x,int fa){//負責找重心 11 siz[x]=1;mx[x]=0; 12 for(int i=h[x],y;i;i=e[i].nxt) 13 if((y=e[i].y)!=fa&&!vis[y]){ 14 getrt(y,x);siz[x]+=siz[y]; 15 mx[x]=max(mx[x],siz[y]); 16 } mx[x]=max(mx[x],sm-siz[x]); 17 if(mx[x]<mx[rt]) rt=x;return ; 18 } voiddfs(int x,int fa){//負責求點到子樹中點的距離 19 rem[++rem[0]]=d[x]; 20 for(int i=h[x],y;i;i=e[i].nxt) 21 if((y=e[i].y)!=fa&&!vis[x]) 22 d[y]=d[x]+e[i].z,dfs(y,x); 23 } void calc(int x){ int p=0;//負責處理x點為根的子樹 24 for(int i=h[x],y;i;i=e[i].nxt) 25 if(!vis[y=e[i].y]){ 26 rem[0]=0;d[y]=e[i].z;dfs(y,x); 27 for(int j=rem[0];j;j--) 28 for(int k=1;k<=m;k++) 29 if(que[k]>=rem[j]) 30 t[k]|=jd[que[k]-rem[j]]; 31 for(int j=rem[0];j;j--) 32 q[++p]=rem[j],jd[rem[j]]=1; 33 } for(int i=1;i<=p;i++) jd[q[i]]=0; 34 } void solve(int x){//負責每次找重心,層層分治 35 vis[x]=jd[0]=1;calc(x); 36 for(int i=h[x],y;i;i=e[i].nxt){ 37 if(vis[y=e[i].y]) continue; 38 sm=siz[y];mx[rt=0]=inf; 39 getrt(y,0);solve(rt); 40 } return; 41 } int main(){ 42 scanf("%d%d",&n,&m); 43 for(int i=1,x,y,z;i<n;i++) 44 scanf("%d%d%d",&x,&y,&z),add(x,y,z); 45 for(int i=1;i<=m;i++) scanf("%d",&que[i]); 46 mx[rt]=sm=n;getrt(1,0);solve(rt); 47 for(int i=1;i<=m;i++) 48 if(t[i]) puts("AYE");else puts("NAY"); 49 return 0; 50 }
Luogu P3806 點分治模板1