《算法之美》-- DFS--部分和&水洼数目
- DFS考虑的穷尽所有
- 考虑状态的变化
- 如果在同一层,是否有些参数存在变化,需要进行回溯
部分和
/* 给定整数序列a1,a2,...,an,判断是否可以从中选出若干数,使它们的和恰好为k. 1≤n≤20 -10^8≤ai≤10^8 -10^8≤k≤10^8 样例: 输入 n=4 a={1,2,4,7} k=13 输出: Yes (13 = 2 + 4 + 7) */
public class Dfs_2部分和 {
private static int kk;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] a = new int[n];
for (int i = 0; i < n; i++) {
a[i] = sc.nextInt();
}
int k = sc.nextInt();//13
kk = k;
dfs(a, k, 0, new ArrayList<Integer>());
}
private static void dfs(int[] a, int k, int cur, ArrayList<Integer> ints) {
if (k == 0) {
System.out.print("Yes (" + kk + " = ");
int size = ints.size();
for (int i = 0; i < size; i++) {
System.out.print(ints.get(i) + (i == size - 1 ? "" : " + "));
}
System.out.println(")");
exit(0);
}
if (k < 0 || cur == a.length) return;
dfs(a, k, cur + 1, ints);//不要cur这个元素
ints.add(a[cur]);
int index = ints.size() - 1;
dfs(a, k - a[cur], cur + 1, ints);
ints.remove(index);//回溯
}
}
水洼数目
/* 有一个大小为 N*M 的园子,雨后积起了水。八连通的积水被认为是连接在一起的。请求出 园子里总共有多少水洼?(八连通指的是下图中相对 W 的*的部分) *** *W* *** 限制条件 N, M ≤ 100 样例: 输入 N=10, M=12 园子如下图('W'表示积水, '.'表示没有积水) W........WW. .WWW.....WWW ....WW...WW. .........WW. .........W.. ..W......W.. .W.W.....WW. W.W.W.....W. .W.W......W. ..W.......W. 输出 3 */
public class Dfs_3水洼数目 {
private static int n;
private static int m;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
char[][] a = new char[n][];
for (int i = 0; i < n; i++) {
a[i] = sc.next().toCharArray();
}
int cnt = 0;
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (a[i][j] == 'W') {
dfs(a, i, j);//清除一个水洼
cnt++;
}
}
}
System.out.println(cnt);
}
private static void dfs(char[][] a, int i, int j) {
a[i][j] = '.';
for (int k = -1; k < 2; k++) {//-1,0,1
for (int l = -1; l < 2; l++) {//-1,0,1
if (k == 0 && l == 0) continue;
if (i + k >= 0 && i + k <= n - 1 && j + l >= 0 && j + l <= m - 1) {
if (a[i + k][j + l] == 'W')
dfs(a, i + k, j + l);
}
}
}
}
}
- 需要避免循环,所以每走一个就需要将其变成没有水