算法笔记第一天-bfs

第一天

第一题：递归实现组合型枚举

#include<iostream>
using namespace std;
const int N=30;

int n,m;
int a[N];
bool mark[N];

void dfs(int x){
    if(x==m){//结束条件
        for(int i=0;i<m;i++)cout<<a[i]<<" ";
        cout<<endl;
        return;
    }
    for(int i=1;i<=n;i++){//递归体
        if(!mark[i]&&a[x-1]<i){
            mark[i]=1;
            a[x]=i;
            dfs(x+1);
            a[x]=0;
            mark[i]=0;
        }
    }
}

int main(){
    cin>>n>>m;
    dfs(0);
    return 0;
}

疑问：当第一次进入dfs时下标不是越界了吗,不明白

第二题：红与黑

#include<iostream>
#include<queue>
using namespace std;
const int N=25;
int ox[4]={-1,0,1,0};
int oy[4]={0,1,0,-1};
int n,m;
char a[N][N];
queue<pair<int,int>>ans;
int res;
int main(){
    while(cin>>m>>n){
        if(!n&&!m)return 0;
        res=0;
        for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
        {
            cin>>a[i][j];
            if(a[i][j]=='@'){
                ans.push({i,j});
                a[i][j]='#';
                res++;
            }
        }
        while(ans.size()){
            auto t=ans.front();
            int x=t.first,y=t.second;
            ans.pop();
            for(int i=0;i<4;i++){
                int cx=x+ox[i],cy=y+oy[i];
                if(cx>=0&&cx<n&&cy>=0&&cy<m&&a[cx][cy]=='.'){
                        a[cx][cy]='#';
                        ans.push({cx,cy});
                        res++;
                }
            }
        }
        cout<<res<<endl;
    }
    return 0;
}

经典bfs

第三题：机器人的运动范围

bool check(int a,int b,int k){
    int ans=0;
    while(a)ans+=a%10,a/=10;
    while(b)ans+=b%10,b/=10;
    return ans<=k;
}

class Solution {
public:
    int movingCount(int threshold, int rows, int cols)
    {
        queue<pair<int,int>>ans;
        bool a[55][55];
        for(int i=0;i<rows;i++){
            for(int j=0;j<cols;j++){
                a[i][j]=0;
            }
        }
        if(!rows&&!cols)return 0;
        a[0][0]=1;
        int res=1;
        int ox[4]={-1,0,1,0};
        int oy[4]={0,1,0,-1};
        ans.push({0,0});
        while(ans.size()){
            auto t=ans.front();
            int x=t.first,y=t.second;
            ans.pop();
            for(int i=0;i<4;i++){
                int cx=x+ox[i],cy=y+oy[i];
                if(cx>=0&&cx<rows&&cy>=0&&cy<cols){
                    if(a[cx][cy]==0&&check(cx,cy,threshold)){
                        a[cx][cy]=1;
                        res++;
                        ans.push({cx,cy});
                    }
                }
            }
        }
        return res;
    }
};

第四题：递归实现排列类型枚举

#include<iostream>
#include<algorithm>
using namespace std;
const int N=10;
int a[N],ans[N];
bool mark[N];
int n;

void dfs(int x){
    if(x==n){
        for(int i=0;i<n;i++)cout<<ans[i]<<" ";
        putchar(10);
        return;
    }
    for(int i=0;i<n;i++){
        if(!mark[i]){
            mark[i]=1;
            ans[x]=a[i];
            dfs(x+1);
            mark[i]=0;
            while(i+1<n&&a[i]==a[i+1])i++;
        }
       
    }
}

int main(){
    cin>>n;
    for(int i=0;i<n;i++)cin>>a[i];
    sort(a,a+n);
    dfs(0);
    return 0;
}

这题数组中的数字可能会出现重复，并且题目要求不能重复，所以要进行判重操作，为了便于判重，先对数组进行排序，然后在if里面，如果在dfs之后下一个数组元素和上次dfs的数相同，那么就跳过。

第五题：小猫爬山

#include<iostream>
using namespace std;
const int N=20;

int a[N],sum[N];
int n,w;
int ans=N;
void dfs(int x,int k){
    if(k>=ans)return;
    if(x==n){
        ans=k;
        return;
    }
    for(int i=0;i<k;i++){
        if(a[x]+sum[i]<=w){
            sum[i]+=a[x];
            dfs(x+1,k);
            sum[i]-=a[x];
        }
    }
    sum[k]=a[x];
    dfs(x+1,k+1);
    sum[k]=0;
    
}

int main(){
    cin>>n>>w;
    for(int i=0;i<n;i++)cin>>a[i];
    dfs(0,0);
    cout<<ans;
    return 0;
}

第六题：火车进站

#include<iostream>
#include<stack>
#include<queue>
using namespace std;
const int N=25;
int a[N];
bool mark[N];
stack<int> st;
vector<int> ans;
int n,cnt=20;
int c=1;

void dfs(){
    if(!cnt)return;
    if(ans.size()==n){
        for(auto i:ans)cout<<i;
        putchar(10);
        cnt--;
        return;
    }
    if(st.size()){
        ans.push_back(st.top());
        st.pop();
        dfs();
        st.push(ans.back());
        ans.pop_back();
    }
    if(c<=n){
    st.push(c);
    c++;
    dfs();
    c--;
    st.pop();
    }
    
    
}

int main(){
    cin>>n;
    dfs();
    return 0;
}

注意，这题不是求全排列。应该根据实际情况进行模拟，得到dfs的规律。

这题我没有加vector的头文件，只加了queue，但还是过了。不可思议

第七题：n皇后

#include<iostream>
#include<cstring>
using namespace std;
const int N=10;
bool a[N][N];
int n;
bool col[N],l[N],r[N];
void dfs(int x){
    if(x==n){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(a[i][j])cout<<'Q';
                else cout<<'.';
            }
            putchar(10);
        }
        putchar(10);
        return;
    }
    for(int i=0;i<n;i++){
        if(!col[i]&&!l[n-x+i]&&!r[x+i]){
            col[i]=l[n-x+i]=r[x+i]=1;
            a[x][i]=1;
            dfs(x+1);
            a[x][i]=0;
            col[i]=l[n-x+i]=r[x+i]=0;
        }
    }
}

int main(){
    cin>>n;
    dfs(0);
    return 0;
}

按行枚举，主要是找到规律，如何将棋盘规则映射到代码上，就是l[]和r[]是怎么算的。

第八题：取石子游戏

#include<iostream>
using namespace std;
int a,b;

bool check(int a,int b){
    if(a/b>=2||a==b)return 1;
    return !check(b,a-b);
}

int main(){
    while(cin>>a>>b){
        if(!a&&!b)break;
        if(a<b)swap(a,b);
        if(check(a,b))cout<<"win"<<endl;
        else cout<<"lose"<<endl;
    }    
    return 0;
}

还是那句话，找规律即可。

第九题：棋盘问题

#include<iostream>
#include<cstring>
using namespace std;
const int N=10;
bool col[N];
char a[N][N];
int n,k;
int ans,m;


//因为每一行可以放也可以不放，所以需要有当前枚举的行数和当前枚举的棋子数两个参数，不能将两个参数合并成一个参数。
void dfs(int x){
    if(m==k){
        ans++;
        return;
    }
    if(x==n)return;
    //按行枚举
    for(int i=0;i<n;i++){
        if(!col[i]&&a[x][i]=='#'){
            col[i]=1;
            m++;
            dfs(x+1);
            m--;
            col[i]=0;
        }
    }
    dfs(x+1);
}

int main(){
    while(cin>>n>>k){
        if(n==-1&&k==-1)break;
        for(int i=0;i<n;i++)cin>>a[i];
        dfs(0);
        cout<<ans<<endl;
        ans=0;
    }
    return 0;
}

需要注意棋子可以放也可以不放。

第十题：棋盘挑战

#include<iostream>
using namespace std;
const int N=15;
int a[N];
bool col[N],l[N<<1],r[N<<1];
int n;
int cnt=3,ans;

void dfs(int x){
    if(x==n){
        ans++;
        if(cnt){
            for(int i=0;i<n;i++)cout<<a[i]<<" ";
            cnt--;
            putchar(10);
        }
        return;
    }
    for(int i=0;i<n;i++){
        if(!col[i]&&!l[i+x]&&!r[n-x+i]){
            col[i]=l[i+x]=r[n-x+i]=1;
            a[x]=i+1;
            dfs(x+1);
            a[x]=0;
            col[i]=l[x+i]=r[n-x+i]=0;
        }
    }
}

int main(){
    cin>>n;
    dfs(0);
    cout<<ans;
    return 0;
}

这题和n皇后一样，需要注意的是，只需输出字典序的前三种方案即可。

在开数组的时候，斜边的数组需要开双倍。，这题的数据范围与n皇后问题不同，这题当n为13时，如果数组不开双倍，就得不到全部的情况，n皇后问题的数据小，不用开也可以过，这里的n皇后问题指的是这个笔记中的n皇后。