《算法之美》-- 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);
        }
      }
    }
  }
}

• 需要避免循环，所以每走一个就需要将其变成没有水