搜索(剪枝matter!)
1. 宽度优先搜索(BFS)
宽度优先搜索,一般是用队列来实现,特别适合求最短时间,最短路径之内的问题,在求解的时候要用一个数据结构去保存以前已经访问的单位避免重复搜索,同时还需要注意剪枝。
1.1 例子
- BFS模版
void BFS(int s){
queue<int> q;
q.push(s);
while(!q.empty()){
取出队首元素top;
访问队首元素top;
将队首元素出队;
将top的下一层结点中<未曾入队>的结点全部入队,并设置为已入队
}
}
- 数块问题
#include<cstdio>
#include<queue>
using namespace std;
const int maxn = 100;
struct node{
int x, y;
}Node;
int n, m;
int matrix[maxn][maxn];
bool inq[maxn][maxn] = {false};
int X[4] = {0, 0, 1, -1};
int Y[4] = {1, -1, 0, 0};
bool judge(int x, int y){
if(x >= n || x < 0 || y >= m || y < 0){
return false;
}
if(matrix[x][y] == 0 || inq[x][y] == true){
return false;
}
return true;
}
void BFS(int x, int y){
queue<node> Q;
Node.x = x, Node.y = y;
Q.push(Node);
inq[x][y] = true;
while(!Q.empty()){
node top = Q.front();
Q.pop();
for(int i = 0; i < 4; i++){
int newX = top.x + X[i];
int newY = top.y + Y[i];
if(judge(newX, newY)){
Node.x = newX, Node.y = newY;
Q.push(Node);
inq[newX][newY] = true;
}
}
}
}
int main(){
scanf("%d%d", &n, &m);
for(int x = 0; x < n; x++){
for(int y = 0;y < m; y++){
scanf("%d", &matrix[x][y]);
}
}
int ans = 0;
for(int x = 0; x < n; x++){
for(int y = 0; y < m; y++){
if(matrix[x][y] == 1 && inq[x][y] == false){
ans++;
BFS(x, y);
}
}
}
printf("%d\n", ans);
return 0;
}
###1.3 注意点
C++STL的queue中,元素入队的push操作只是制造了该元素的一个副本入队,入队后队原来元素的修改不会影响队列中的副本,对队列中副本的修改也不改变原元素。
为了将上述两者关联,一种可行的做法就是在queue中保存下标。
1.4 应用练习
2. 深度优先搜索(DFS)
深度优先搜索,一般使用递归来实现(系统栈),当然深度优先搜索也可以通过STL栈实现
2.1例子
- 背包问题. 神奇的口袋(王道P152)
void DFS(int index, int sumW, int sumC){
if(index == n){
return;
}
DFS(index+1,sumW, sumC);
if(sumW + w[index] <= V){
if(sumC + c[index] > ans){
ans = sumC + c[index];
}
DFS(index+1, sumW + w[index], sumC + c[index])
}
}
- N选K个数
void DFS(int index, int nowK(总共k个数), int sum(和为b), int sumSqu(最大))
2.2 应用练习
[A Knight’s Journey(王道P146)]