BZOJ2783: [JLOI2012]樹
阿新 • • 發佈:2018-08-13
size key tro con 過程 所有權 orange 超過 sample
題解: 不得不吐槽bzoj的題意看的一臉懵逼啊 轉站洛谷讀了一發題 ....很裸 dfs過程中維護前綴和 二分一下是否會有滿足的位置即可
#include <bits/stdc++.h> const int MAXN=1e5+10; using namespace std; vector<int>vec[MAXN]; int a[MAXN],num[MAXN],key; bool vis[MAXN];int n,m; int slove(int l,int r,int t){ if(l>r)return 0; int pos=r,ans=0; while(l<=r){ int mid=(l+r)>>1; if(num[pos]-num[mid-1]>=t)ans=num[pos]-num[mid-1],l=mid+1; else r=mid-1; } if(ans==t)return 1; return 0; } void dfs(int v,int pre,int dep){ if(m!=a[v])key+=slove(1,dep,m-a[v]); else key++; num[dep+1]=num[dep]+a[v]; for(int i=0;i<vec[v].size();i++){ if(vec[v][i]!=pre){ dfs(vec[v][i],v,dep+1); } } } int main(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;i++)scanf("%d",&a[i]); int u,v; for(int i=1;i<n;i++){ scanf("%d%d",&u,&v); vec[u].push_back(v);vis[v]=1; } int S;key=0; for(int i=1;i<=n;i++)if(!vis[i]){S=i;break;} dfs(S,0,1); if(a[S]==m)key++; printf("%d\n",key); }
題目描述
在這個問題中,給定一個值S和一棵樹。在樹的每個節點有一個正整數,問有多少條路徑的節點總和達到S。路徑中節點的深度必須是升序的。假設節點1是根節點,根的深度是0,它的兒子節點的深度為1。路徑不必一定從根節點開始。
輸入輸出格式
輸入格式:第一行是兩個整數N和S,其中N是樹的節點數。 第二行是N個正整數,第i個整數表示節點i的正整數。 接下來的N-1行每行是2個整數x和y,表示y是x的兒子。
輸出格式:輸出路徑節點總和為S的路徑數量。
輸入輸出樣例
輸入樣例#1: 復制3 3 1 2 3 1 2 1 3輸出樣例#1: 復制
2
說明
對於100%數據,N<=100000,所有權值以及S都不超過1000。
BZOJ2783: [JLOI2012]樹