1. 程式人生 > >BZOJ2783: [JLOI2012]樹

BZOJ2783: [JLOI2012]樹

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]樹