《算法之美》-- 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);
        }
      }
    }
  }
}
  • 需要避免循环,所以每走一个就需要将其变成没有水
全部评论

相关推荐

11-24 11:23
门头沟学院 C++
点赞 评论 收藏
分享
双非一本失业第二年:《机器视觉垃圾分类》
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务