牛牛打怪兽(C++)---DFS解题

 

题目描述

题意

身为屯里第一剑士的牛牛来到训练场里闯关，由于过于勤奋，牛牛的宝剑的耐久度降到了 222 ，这意味着牛牛最多只能打倒两只怪兽，否则将会被淘汰。

训练场的地图可以看作一棵以 111 为根节点的树，训练场的终点为这棵树的叶子结点，树上的每个结点最多有一只怪兽，结点与结点间的边上没有怪兽。

每一个有怪兽的结点上牛牛都需要打倒怪兽才算安全，并且牛牛一旦选定好打怪路线之后便不能走回头路。

请问牛牛有多少种到达终点且不被淘汰的路径。

输入

第一个参数为 nnn ，(1≤n≤100,000)(1\leq n\leq 100,000)(1≤n≤100,000)

第二个参数为大小为 n−1n-1n−1 的点对 (ui,vi)(u_i, v_i)(ui​,vi​) 的集合，其中 (ui,vi)(u_i, v_i)(ui​,vi​) 表示结点 uiu_iui​ 与结点 viv_ivi​ 之间有一条边，1≤ui,vi≤n1\leq u_i, v_i \leq n1≤ui​,vi​≤n

第三个参数为大小为 nnn 的 0/10/10/1 序列 fff ，若 fif_ifi​ 为 000 表示i-1结点没有怪兽，否则表示 i-1 结点有怪兽。

返回

一个整数，表示牛牛能到达终点且不被淘汰的路径数。

示例1

输入

7,[(7,2),(6,1),(5,2),(1,2),(4,6),(6,3)],[0,0,1,0,1,0,0]

输出

4

说明

样例中的四条路径分别为: (1 - 2 - 7), (1 - 2 - 5) , (1 - 6 - 3), (1 - 6 - 4)

 

 

---

 

分析

先构建邻接矩阵edges，然后直接调用dfs求出ans。
自我感觉难点在找叶子结点上，其实可以通过找当前节点 u 是否有子节点来判断：如果没有子节点，也就是 u 连接的只有父节点，那么可以判断 u 为叶子节点；如果有子节点，也就是 u 连接的除了有父节点 还有子节点，那么可以判断 u 不是叶子节点。



-解题代码 1

/**
 * struct Point {
 *	int x;
 *	int y;
 * };
 */

class Solution {
public:
	int ans = 0;
	vector> edges;
	
	void dfs(const vector& f, int u, int fa, int heath) {
		heath -= f[u];
		if (heath < 0) return ;
		
		bool leaf = true;
		for(int v : edges[u]) {
			if (v != fa) { //说明 u 有子节点，也就是 u 不是叶子结点 
				dfs(f, v, u, heath);
				leaf = false;
			}
		}
		
		if (leaf) ans++;
	}
	
    int solve(int n, vector& Edge, vector& f) {
        edges.resize(n);
        for(const Point& pt : Edge) {
        	edges[pt.x - 1].push_back(pt.y - 1);
        	edges[pt.y - 1].push_back(pt.x - 1);
        }
        dfs(f, 0, -1, 2);
        return ans;
    }
};

-解题代码 2

/**
 * struct Point {
 *	int x;
 *	int y;
 * };
 */

class Solution {
public:
	int ans = 0;
	vector> edges;
	
	void dfs(const vector& f, int u, int fa, int heath) {
		heath -= f[u];
		if (heath < 0) return ;
		
		int k = 0; //k 为 u 的子节点个数 
		for(int v : edges[u]) {
			if (v == fa) continue;
			dfs(f, v, u, heath);
			k++;
			
		}
		
		if (k == 0) ans++;
	}
	
    int solve(int n, vector& Edge, vector& f) {
        edges.resize(n);
        for(const Point& pt : Edge) {
        	edges[pt.x - 1].push_back(pt.y - 1);
        	edges[pt.y - 1].push_back(pt.x - 1);
        }
        dfs(f, 0, -1, 2);
        return ans;
    }
};