分类 深度优先搜索 下的文章

组合

问题描述

给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

思路

回溯算法

题解

class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> ret;
    vector<vector<int>> combine(int n, int k) {
        dfs(n,k,1);
        return ret;
    }
    void dfs(int n,int k,int location){
        if(tmp.size()==k){
            ret.push_back(tmp);
            return ;
        }
        for(int i=location;i<=n;i++){
            tmp.push_back(i);
            dfs(n,k,i+1);
            tmp.pop_back();
        }
    }
};

二叉树的前序遍历

问题描述

给定一个二叉树,返回它的 前序 遍历。

示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [1,2,3]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

思路

深度优先搜索。

题解

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> ret;
    vector<int> preorderTraversal(TreeNode* root) {
        dfs(root);
        return ret;
    }
    void dfs(TreeNode *node){
        if(node==nullptr){
            return ;
        }
        ret.push_back(node->val);
        dfs(node->left);
        dfs(node->right);
    }
};

组合总和

问题描述

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。

说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
 
示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例 2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum

思路

本题用回溯算法,思路比较简单。
要解决的问题是,从组合中选取几个数,使得和为target。思路和以前的题没啥太大的区别,这里就不多做赘述了。
需要注意的两点是,1、没有个数的限制,只需要保证sum==target。所以,当深度达到sum>=target时,返回到上一层。这里同样需要注意,返回到上一层后将tmp.pop_back();将sum-=candidates[i];

题解

class Solution {
public:
    vector<int> tmp;
    vector<vector<int> > ret;
    int n;
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        n=candidates.size();
        dfs(candidates,target,0,0);
        return ret;
    }
    void dfs(vector<int> &candidates,int target,int sum,int location){
       if(sum>target){
           return;
       } else if(sum==target){
           ret.push_back(tmp);
           return;
       }
       for(int i=location;i<n;i++){
           tmp.push_back(candidates[i]);
           sum+=candidates[i];
           dfs(candidates,target,sum,i);
           tmp.pop_back();
           sum=sum-candidates[i];
       }
    }
};

二叉树的中序遍历

问题描述

给定一个二叉树,返回它的中序 遍历。

示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]

进阶: 递归算法很简单,你可以通过迭代算法完成吗?

思路

深度优先搜索

题解

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> ret;
    vector<int> inorderTraversal(TreeNode* root) {
        dfs(root);
        return ret;
    }
    void dfs(TreeNode *node){
        if(node==nullptr){
            return ;
        }
        dfs(node->left);
        ret.push_back(node->val);
        dfs(node->right);
    }
};

全排列 II

问题描述

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

思路

  • 回溯算法,得到所有的排序情况ret;
  • 对ret进行排序,使用ret.erase(unique(ret.begin(), ret.end()), ret.end())对ret进行去重

题解

class Solution {
public:
    vector<vector<int>> ret;
    vector<int> tmp;
    int n;
    bool visited[100] = {false};

    vector<vector<int>> permuteUnique(vector<int> &nums) {
        n = nums.size();
        if (n == 0) return ret;
        dfs(nums);
        sort(ret.begin(), ret.end());
        ret.erase(unique(ret.begin(), ret.end()), ret.end());
        return ret;
    }

    void dfs(vector<int> nums) {
        if (tmp.size() == n) {
            ret.push_back(tmp);
            return;
        }
        for (int i = 0; i < n; i++) {
            if (visited[i] == false) {
                visited[i] = true;
                tmp.push_back(nums[i]);
                dfs(nums);
                tmp.pop_back();
                visited[i] = false;
            }
        }
    }
};