dfs的姿势 尽量减少不必要的搜索

好长时间不做搜索了 ,虽然做了一个简单的果断tle 和别人代码一比就看出高低了 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <iostream>
#include <sstream>
#include <ostream>
#include <algorithm>
#include <ctype.h>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <vector>
#define inf 1e9+7
#define pi acos(-1)
#define natrule exp(1)
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000")
int mat[10][10];
int marklines[10];
int ans=0;
void dfs(int n,int k,int row){
    if(k==0) {ans++;return ;}
    if(n-row+1<k) return;
    dfs(n,k,row+1);//这一行不放 每一行的状态放与不放 这样搜索的少一点 不用强制循环搜索这么多
    for(int i=1;i<=n;i++){
        if(mat[row][i]&&marklines[i]==0) {
            marklines[i]=1;
            dfs(n,k-1,row+1);
            marklines[i]=0;
        }
    }
}
int main()
{
    int n,k;
    while(cin>>n>>k){
        if(n==-1&&k==-1) break;
        memset(marklines,0,sizeof(marklines));
        char s[20];
        ans=0;
        int cnt=0;
        for(int i=1;i<=n;i++){
            cin>>s;
            for(int j=1;j<=n;j++){
                if(s[j-1]=='#') {mat[i][j]=1;cnt++;}
                else mat[i][j]=0;
            }
        }
        if(cnt<k) {cout<<0<<endl;continue;}
        dfs(n,k,1);
        cout<<ans<<endl;
    }
    return 0;
}

这个对于行数进行分类 : 选择这行或者是不选择这一行

其实想想很有道理 本来就是np问题 这个样子之后就相当于做判断 这一行选不选

而我把所有位置都搜一变 复杂度已经变成了 (n^2)^k肯定超时 下面是教训代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <iostream>
#include <sstream>
#include <ostream>
#include <algorithm>
#include <ctype.h>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <vector>
#define inf 1e9+7
#define pi acos(-1)
#define natrule exp(1)
using namespace std;
#pragma comment(linker, "/STACK:1024000000,1024000000")
int mat[10][10];
int markrows[10];
int marklines[10];
int ans=0;
void dfs(int n,int k,int row){
    if(k==0) {ans++;return ;}
    for(int i=row;i<=n;i++){
        if(n-row+1<k) break;
        for(int j=1;j<=n;j++){
            if(mat[i][j]&&!marklines[j]&&!markrows[i]){
                mat[i][j]=0;
                marklines[j]=1;
                markrows[i]=1;
                dfs(n,k-1,row+1);
                marklines[j]=0;
                markrows[i]=0;
                mat[i][j]=1;
            }
        }
    }
}
int main()
{
    int n,k;
    while(cin>>n>>k){
        if(n==-1&&k==-1) break;
        memset(marklines,0,sizeof(marklines));
        memset(markrows,0,sizeof(markrows));
        char s[20];
        ans=0;
        int cnt=0;
        for(int i=1;i<=n;i++){
            cin>>s;
            for(int j=1;j<=n;j++){
                if(s[j-1]=='#') {mat[i][j]=1;cnt++;}
                else mat[i][j]=0;
            }
        }
        if(cnt<k) {cout<<0<<endl;continue;}
        dfs(n,k,1);
        cout<<ans<<endl;
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值