Kefa and Park

题意（Kefa and Park）:

        给出n个点及相应长度序列表示点的状态（0 或1）， 求从根节点走到叶子路径（路径中不存在连续k 个1）数。

思路：

        题目保证树故用dfs 搜索，当成无向图存储， （如 只有一条边2， 1 时有向图存储会导致错误）重点在于叶子结点的判断， 发现叶子结点e[u].size() == 1, 但非叶子结点（如根节点也可能子树为一）， 故增加一个是否存在子树的判断。

代码：

        

 #include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define PII pair<int, int>
#define PLL pair<long, long>
typedef long long ll;
const int N = 1e5 + 7;
#define lowbit(x) (x & (-x))
ll rs(ll p) {return p << 1 | 1;}
ll ls(ll p) {return p << 1;}
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
int n, m;
int cat[N];
ll ans;
vector<int> e[N];
int cnt = 0;
void dfs(int u, int fa, int sum) {
	if (cat[u] == 0) sum = 0;
	else sum++;
	if (sum > m) return;
	bool f = false;
	for (int &v : e[u]) {
		if (v == fa) continue;
		dfs(v, u, sum);
		f = true;//f = true 说明存在子树
	}
	if (!f && e[u].size() == 1) ans++;        
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++){
		cin >> cat[i];
	}
	for (int i = 1; i < n; i++) {
		int u, v; cin >> u >> v;
		e[u].push_back(v);
		e[v].push_back(u);
	}  
	dfs(1, 0, 0);
	cout << ans;
	return 0;
}