DFS
733. 图像渲染
https://leetcode.cn/problems/flood-fill/
有一幅以 m x n 的二维整数数组表示的图画 image ,其中 image[i] [j] 表示该图画的像素值大小。
你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[sr] [sc] 开始对图像进行 上色填充 。
为了完成 上色工作 ,从初始像素开始,记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为 newColor 。
最后返回 经过上色渲染后的图像 。
class Solution {
public:
vector<vector<int>> g;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
void dfs(int x, int y, int color, int newColor){
g[x][y] = newColor;
for(int i = 0; i < 4; i++){
int a = x + dx[i];
int b = y + dy[i];
if(a >= 0 && b >= 0 && a < g.size() && b < g[0].size() && g[a][b] == color){
dfs(a, b, color, newColor);
}
}
}
vector<vector<int>> floodFill(vector<vector<int>>& image, int x, int y, int newColor) {
g = image;
if(g[x][y] == newColor) return g;
dfs(x, y, g[x][y], newColor);
return g;
}
};
难点在于模拟上下左右四个方向,其他的不算困难,板子题。
695. 岛屿的最大面积
https://leetcode.cn/problems/max-area-of-island/submissions/
给你一个大小为 m x n 的二进制矩阵 grid 。
岛屿 是由一些相邻的 1 (代表土地) 构成的组合,这里的「相邻」要求两个 1 必须在 水平或者竖直的四个方向上 相邻。你可以假设 grid 的四个边缘都被 0(代表水)包围着。
岛屿的面积是岛上值为 1 的单元格的数目。
计算并返回 grid 中最大的岛屿面积。如果没有岛屿,则返回面积为 0 。
class Solution {
public:
vector<vector<int>> g;
int n, m;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
int dfs(int x, int y){
int res = 1;
g[x][y] = 0;
for(int i = 0; i < 4; i++){
int a = x + dx[i], b = y + dy[i];
if(a >= 0 && b >= 0 && a < n && b < m && g[a][b]){
res += dfs(a, b);
}
}
return res;
}
int maxAreaOfIsland(vector<vector<int>>& grid) {
g = grid;
n = g.size(), m = g[0].size();
int res = 0;
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
if(g[i][j]){
res = max(res, dfs(i, j));
}
}
}
return res;
}
};
注意上下左右的偏移量操作。
617. 合并二叉树
https://leetcode.cn/problems/merge-two-binary-trees/
给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root2) swap(root1, root2);
if(!root1) return NULL;
if(root2) root1->val += root2->val;
root1->left = mergeTrees(root1->left, root2 ? root2->left : NULL);
root1->right = mergeTrees(root1->right, root2 ? root2->right : NULL);
return root1;
}
};
先确保 root1 肯定不为空,要不然直接返回 NULL,接着直接递归深搜左右两边即可。
77. 组合
https://leetcode.cn/problems/combinations/
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> combine(int n, int k) {
dfs(n, k, 1);
return ans;
}
void dfs(int n, int k, int start){
if(!k){
ans.push_back(path);
return;
}
for(int i = start; i <= n; i++){
path.push_back(i);
dfs(n, k - 1, i + 1);
path.pop_back();
}
}
};
递归一下即可。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combine(int n, int k) {
dfs(n, k, 1);
return res;
}
void dfs(int n, int k, int start){
if(path.size() == k) {
res.push_back(path);
return;
}
for(int i = start; i <= n; i++) {
path.push_back(i);
dfs(n, k, i + 1);
path.pop_back();
}
}
};
更易懂的写法。
46. 全排列
https://leetcode.cn/problems/permutations
给定一个不含重复数字的数组 nums
,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
class Solution {
public:
vector<vector<int>> ans;
vector<bool> st;
vector<int> path;
vector<vector<int>> permute(vector<int>& nums) {
path = vector<int>(nums.size());
st = vector<bool>(nums.size());
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u){
if(u == nums.size()){
ans.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i++){
if(st[i] == false){
st[i] = true;
path[u] = nums[i];
dfs(nums, u + 1);
st[i] = false;
}
}
}
};
直接爆搜,注意存放状态的问题。
784. 字母大小写全排列
https://leetcode.cn/problems/letter-case-permutation/
给定一个字符串 s
,通过将字符串 s
中的每个字母转变大小写,我们可以获得一个新的字符串。
返回 所有可能得到的字符串集合 。以 任意顺序 返回输出。
class Solution {
public:
vector<string> ans;
vector<string> letterCasePermutation(string s) {
dfs(s, 0);
return ans;
}
void dfs(string &s, int u){
if(u == s.length()){
ans.push_back(s);
return;
}
dfs(s, u + 1);
if (s[u] >= 'A')
{
s[u] ^= 32;
dfs(s, u + 1);
s[u] ^= 32;
}
}
};
异或运算直接用 ^= 32
来切换大小写,爆搜即可。
216. 组合总和 III
https://leetcode.cn/problems/combination-sum-iii/
找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字 1 到 9 每个数字 最多使用一次 返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combinationSum3(int k, int n) {
dfs(n, k, 1);
return res;
}
void dfs(int n, int k, int start){
if(!n) {
if(!k) {
res.push_back(path);
return;
} else return;
} else if (k) {
for(int i = start; i <= 9; i++){
if(n >= i) {
path.push_back(i);
dfs(n - i, k - 1, i + 1);
path.pop_back();
}
}
} else return;
}
};
组合题必须有一个顺序,这里用 start 开始即可,注意每一层的条件判断。
17. 电话号码的字母组合
https://leetcode.cn/problems/letter-combinations-of-a-phone-number/
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
class Solution {
public:
vector<string> ans;
string strs[10] = {
"", "", "abc", "def",
"ghi", "jkl", "mno",
"pqrs", "tuv", "wxyz",
};
vector<string> letterCombinations(string digits) {
if(!digits.size()) return {};
dfs(digits, 0, "");
return ans;
}
void dfs(string &digits, int u, string path){
if(u == digits.size()) {
ans.push_back(path);
return;
}
for(auto c : strs[digits[u] - '0']) {
dfs(digits, u + 1, path + c);
}
}
};
难点在于用一个数组来存储每一个数对应的字母。
39. 组合总和
https://leetcode.cn/problems/combination-sum/
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
dfs(candidates, target, 0);
return res;
}
void dfs(vector<int>& candidates, int target, int u){
if(!target) {
res.push_back(path);
return;
}
if(u == candidates.size()) return;
for(int i = 0; target >= candidates[u] * i; i++){
dfs(candidates, target - candidates[u] * i, u + 1);
path.push_back(candidates[u]);
}
for(int i = 0; target >= candidates[u] * i; i++){
path.pop_back();
}
}
};
第一个 for 循环中的 i 为这个数被选中的次数,第二次 for 循环恢复现场。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> combinationSum(vector<int>& nums, int target) {
dfs(nums, target, 0);
return ans;
}
void dfs(vector<int>& nums, int target, int u) {
if(!target) {
ans.push_back(path);
return;
}else if(target < 0) return;
for(int i = u; i < nums.size(); i++) {
target -= nums[i];
path.push_back(nums[i]);
dfs(nums, target, i);
target += nums[i];
path.pop_back();
}
}
};
更好的解法,每次递归时的 i + 1 改成 i,相当于给树多加了一条分支。
40. 组合总和 II
https://leetcode.cn/problems/combination-sum-ii/
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
dfs(candidates, target, 0);
return res;
}
void dfs(vector<int>& candidates, int target, int u) {
if(!target) {
res.push_back(path);
return;
}
if(u == candidates.size()) return;
int k = u + 1;
while(k < candidates.size() && candidates[k] == candidates[u]) k++;
int cnt = k - u;
for(int i = 0; target >= candidates[u] * i && cnt >= i; i++){
dfs(candidates, target - candidates[u] * i, k);
path.push_back(candidates[u]);
}
for(int i = 0; target >= candidates[u] * i && cnt >= i; i++){
path.pop_back();
}
}
};
这里 dfs 第二个参数为 k 实际上就是直接寻找到下个不同的数,因为在循环中已经会把相同的数考虑一遍了。
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
dfs(candidates, target, 0);
return res;
}
void dfs(vector<int>& candidates, int target, int u) {
if(!target) {
res.push_back(path);
return;
} else if(target < 0) return;
for(int i = u; i < candidates.size(); i++){
if(i > u && candidates[i] == candidates[i - 1]) continue;
path.push_back(candidates[i]);
target -= candidates[i];
dfs(candidates, target, i + 1);
path.pop_back();
target += candidates[i];
}
}
};
更好的做法,通过剪枝优化。
131. 分割回文串
https://leetcode.cn/problems/palindrome-partitioning/
class Solution {
public:
vector<vector<bool>> f;
vector<vector<string>> ans;
vector<string> path;
vector<vector<string>> partition(string s) {
int n = s.size();
f = vector<vector<bool>>(n, vector<bool>(n));
for(int j = 0; j < n; j++){
for(int i = 0; i <= j; i++){
if(i == j) f[i][j] = true;
else if(s[i] == s[j]){
if(f[i + 1][j - 1] || i + 1 > j - 1) f[i][j] = true;
}
}
}
dfs(s, 0);
return ans;
}
void dfs(string &s, int n){
if(n == s.size()) ans.push_back(path);
else{
for(int i = n; i < s.size(); i++){
if(f[n][i]) {
path.push_back(s.substr(n, i - n + 1));
dfs(s, i + 1);
path.pop_back();
}
}
}
}
};
这道题首先会通过 f 数组将所有组合是否是回文串存起来,接着进行爆搜即可。
93. 复原 IP 地址
https://leetcode.cn/problems/restore-ip-addresses/
class Solution {
public:
vector<string> ans;
vector<string> restoreIpAddresses(string s) {
dfs(s, 0, 0, "");
return ans;
}
void dfs(string &s, int u, int k, string path){
if(u == s.size()) {
if(k == 4) {
path.pop_back();
ans.push_back(path);
}
return;
}
if(k == 4) return;
for(int i = u, t = 0; i < s.size(); i++) {
if(i > u && s[u] == '0') break;
t = t * 10 + s[i] - '0';
if(t <= 255) dfs(s, i + 1, k + 1, path + to_string(t) + ".");
else break;
}
}
};
掌握 IP 的规则进行暴搜即可。
78. 子集
https://leetcode.cn/problems/subsets/
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> subsets(vector<int>& nums) {
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int n){
if(n <= nums.size()) ans.push_back(path);
if(n > nums.size()) return;
for(int i = n; i < nums.size(); i++){
path.push_back(nums[i]);
dfs(nums, i + 1);
path.pop_back();
}
}
};
如果 u 当前数字小于 nums.size() 长度,那么就保存进结果数组,如果大于的话,直接返回; i = u 开始,i < nums.size(); 递归调用下一个 dfs(i+1)。
90. 子集 II
https://leetcode.cn/problems/subsets-ii/
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u) {
if(u == nums.size()) {
ans.push_back(path);
return;
}
int k = u;
while(k < nums.size() && nums[k] == nums[u]) k++;
dfs(nums, k);
for(int i = u; i < k; i++) {
path.push_back(nums[i]);
dfs(nums, k);
}
for(int i = u; i < k; i++) path.pop_back();
}
};
与上题不同的是本题可以包含重复元素,为了方便处理,我们先将数组排序,这样相同元素就会排在一起。然后暴力搜索所有方案,搜索顺序是这样的:我们先枚举每个不同的数,枚举到数 x 时,我们再求出 x 的个数 k,然后我们枚举在集合中放入 0,1,2,…k 个 x,共 k + 1 种情况。当枚举完最后一个数时,表示我们已经选定了一个集合,将该集合加入答案中即可。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u) {
ans.push_back(path);
for(int i = u; i < nums.size(); i++) {
if(i > u && nums[i] == nums[i - 1]) continue;
path.push_back(nums[i]);
dfs(nums, i + 1);
path.pop_back();
}
}
};
更好理解的做法,通过剪枝来进行优化。
491. 递增子序列
https://leetcode.cn/problems/non-decreasing-subsequences/
给你一个整数数组 nums ,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。
数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> findSubsequences(vector<int>& nums) {
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u) {
if(path.size() >= 2) ans.push_back(path);
if(u == nums.size()) return;
unordered_set<int> S;
for(int i = u; i < nums.size(); i++){
if(path.empty() || path.back() <= nums[i]){
if(S.count(nums[i])) continue;
S.insert(nums[i]);
path.push_back(nums[i]);
dfs(nums, i + 1);
path.pop_back();
}
}
}
};
47. 全排列 II
https://leetcode.cn/problems/permutations-ii/
给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<bool> st;
vector<vector<int>> permuteUnique(vector<int>& nums) {
st = vector<bool>(nums.size());
dfs(nums, 0);
return ans;
}
void dfs(vector<int>& nums, int u) {
if(u == nums.size()) {
ans.push_back(path);
return;
}
unordered_set<int> S;
for(int i = 0; i < nums.size(); i++){
if(!st[i]) {
if(S.count(nums[i])) continue;
S.insert(nums[i]);
st[i] = true;
path.push_back(nums[i]);
dfs(nums, u + 1);
st[i] = false;
path.pop_back();
}
}
}
};
用一个哈希集合来进行判重即可。
79. 单词搜索
https://leetcode.cn/problems/word-search/
给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false 。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
class Solution {
public:
bool exist(vector<vector<char>>& board, string word) {
if(board.empty() || board[0].empty()) return false;
int n = board.size(), m = board[0].size();
for(int i = 0; i < n; i++) {
for(int j = 0; j < m; j++) {
if(dfs(board, word, 0, i, j)) return true;
}
}
return false;
}
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
bool dfs(vector<vector<char>>& board, string word, int u, int x, int y) {
if(board[x][y] != word[u]) return false;
if(u == word.size() - 1) return true;
int n = board.size(), m = board[0].size();
int t = board[x][y];
board[x][y] = '.';
for(int i = 0; i < 4; i++) {
int a = x + dx[i], b = y + dy[i];
if(a < 0 || b < 0 || a >= n || b >= m || board[a][b] == '.') continue;
if(dfs(board, word, u + 1, a, b)) return true;
}
board[x][y] = t;
return false;
}
};
通过深搜,上下左右找即可。
200. 岛屿数量
https://leetcode.cn/problems/number-of-islands/
给你一个由 '1'(陆地)和 '0'(水)组成的的二维网格,请你计算网格中岛屿的数量。
岛屿总是被水包围,并且每座岛屿只能由水平方向和/或竖直方向上相邻的陆地连接形成。
此外,你可以假设该网格的四条边均被水包围。
class Solution {
public:
vector<vector<char>> g;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
int numIslands(vector<vector<char>>& grid) {
g = grid;
int cnt = 0;
for(int i = 0; i < g.size(); i++) {
for(int j = 0; j < g[i].size(); j++) {
if(g[i][j] == '1') {
dfs(i, j);
cnt++;
}
}
}
return cnt;
}
void dfs(int x, int y) {
g[x][y] = 0;
for(int i = 0; i < 4; i++) {
int a = x + dx[i], b = y + dy[i];
if(a >= 0 && b >= 0 && a < g.size() && b < g[a].size() && g[a][b] == '1') dfs(a, b);
}
}
};
经典洪水算法,模版题。
22. 括号生成
https://leetcode.cn/problems/generate-parentheses/
数字 n 代表生成括号的对数,请你设计一个函数,用于能够生成所有可能的并且 有效的 括号组合。
class Solution {
public:
vector<string> generateParenthesis(int n) {
vector<string> res;
string path;
dfs(n, n, res, path);
return res;
}
void dfs(int l, int r, vector<string>& res, string& path) {
if(l < 0 || r < 0) return;
if(l > r) return;
if(l == 0 && r == 0) res.push_back(path);
path.push_back('(');
dfs(l - 1, r, res, path);
path.pop_back();
path.push_back(')');
dfs(l, r - 1, res, path);
path.pop_back();
}
};
4. 寻找两个正序数组的中位数
https://leetcode.cn/problems/median-of-two-sorted-arrays/
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int tot = nums1.size() + nums2.size();
if (tot % 2 == 0) {
int left = find(nums1, 0, nums2, 0, tot / 2);
int right = find(nums1, 0, nums2, 0, tot / 2 + 1);
return (left + right) / 2.0;
} else return find(nums1, 0, nums2, 0, tot / 2 + 1);
}
int find(vector<int>& nums1, int i, vector<int>& nums2, int j, int k) {
if (nums1.size() - i > nums2.size() - j) return find(nums2, j, nums1, i, k);
if (k == 1) {
if (nums1.size() == i) return nums2[j];
else return min(nums1[i], nums2[j]);
}
if (nums1.size() == i) return nums2[j + k - 1];
int si = min((int)nums1.size(), i + k / 2), sj = j + k - k / 2;
if (nums1[si - 1] > nums2[sj - 1]) return find(nums1, i, nums2, sj, k - (sj - j));
else return find(nums1, si, nums2, j, k - (si - i));
}
};
最后更新于