深度优先搜索
经指导,才发现我对深度优先搜索的理解非常有问题,不过发现问题及时改正!
先上深度优先搜索模板,从模板中可以看出,我们需要搜索结束的条件,以及每一次搜索需要作出的选择,作出选择以后继续搜索,搜索完成以后,需要撤销选择,这一步骤很重要,相当于回退到之前搜索过的结点,然后进行下一次的选择。
简而言之,搜索就是在状态空间树中遍历,相当于一个树的遍历。
题目:全排列
输入n个数字,输出这n个数组的全排列。
分析:首先需要一个数据结构来确定还有哪些数据没有被排序,因此有了bool数组is_in,然后有一个数组用来装已经被排列的数组,因此有了数组bucket。
结束条件是当选了n个数字以后,也就是没有数字可以放入bucket的时候,结束深度优先搜索。
选择列表就是剩余未排列的数字,这里需要用is_in来判断是否已经被排列,如果被排列就下一个数字。
代码
#include<iostream>
using namespace std;
const int max_num = 100;
int n;
int ans=0;
int number[max_num];
int bucket[max_num];
bool is_in[max_num];
void dfs(int step)
{
if(step==n+1)
{
for(int j=1;j<=n;j++)
{
cout<<bucket[j]<<' ';
}
cout<<endl;
return ;
}
for(int k=1;k<=n;k++)
{
if(is_in[k])
{
bucket[step]=number[k];
//cout<<"k="<<k<<' '<<number[k]<<endl;
is_in[k]=false;
dfs(step+1);
is_in[k]=true;
}
}
return;
}
int main()
{
for(int i=0;i<max_num;i++)
{
is_in[i]=true;
}
cout<<"请输入需要全排列的数字的个数"<<endl;
cin>>n;
cout<<"请输入全排列的数字"<<endl;
for(int i=1;i<=n;i++)
{
cin>>number[i];
}
dfs(1);
return 0;
}
题目:来自leetcode
链接:https://leetcode-cn.com/problems/target-sum/
给定一个非负整数数组,a1, a2, …, an, 和一个目标数,S。现在你有两个符号 + 和 -。对于数组中的任意一个整数,你都可以从 + 或 -中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
这个题也可以用动态规划来解,等我学到动态规划的时候再重新写一遍
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int S) {
return dfs(nums, S, 0);
}
int dfs(vector<int> &nums, uint target, int left) {
if (target == 0 && left == nums.size()) return 1;
if (left >= nums.size()) return 0;
int ans = 0;
ans += dfs(nums, target - nums[left], left + 1);
ans += dfs(nums, target + nums[left], left + 1);
return ans;
}
};
超时了
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int S) {
return dfs(nums, S, 0);
}
int dfs(vector<int> &nums, uint target, int left) {
if (target == 0 && left == nums.size()) return 1;
if (left >= nums.size()) return 0;
int ans = 0;
target-=nums[left];
ans += dfs(nums, target, left + 1);
target+=nums[left];
target+=nums[left];
ans += dfs(nums, target, left + 1);
target-=nums[left];
return ans;
}
};