poj1321 dfs(bool数组memset更快(比int),15ms左右)

//Time
//32ms
//Memory
//184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
char s[10][10];
int ss[10];
int n,k,num;
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    for(int i=l;i<n;i++)
        for(int j=0;j<n;j++)
    {
        if(ss[j]||s[i][j]!='#')continue;     //这里有一个妙用
        ss[j]=1;dfs(i+1,step+1);ss[j]=0;
    }

}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(s,0,sizeof s);
        memset(ss,0,sizeof ss);
        for(int i=0;i<n;i++)
            scanf("%s",&s[i]);
        dfs(0,0);
            cout<<num<<endl;
    }

}

Time
32ms
Memory
184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
char s[10][10];
bool ss[10];
int n,k,num;
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    for(int i=l;i<n;i++)
        for(int j=0;j<n;j++)
    {
        if(ss[j]||s[i][j]!='#')continue;     //这里有一个妙用
        ss[j]=true;dfs(i+1,step+1);ss[j]=false;
    }

}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(s,0,sizeof s);
        memset(ss,false,sizeof ss);
        for(int i=0;i<n;i++)
            scanf("%s",&s[i]);
        dfs(0,0);
            cout<<num<<endl;
    }

}

Time
47ms
Memory
196kB
#include<iostream>
using namespace std;

bool chess[9][9];
bool vist_col[9];  //列标记
int status;  //状态计数器
int n,k;

void DFS(int row,int num)  //逐行搜索,row为当前搜索行,num为已填充的棋子数
{
	if(num==k)
	{
		status++;
		return;
	}

	if(row>n)    //配合下面DFS(row+1,num); 语句使用,避免搜索越界
		return;
    for(int i=row;i<=n;i++)
	for(int j=1;j<=n;j++)
		if(chess[i][j] && !vist_col[j])
		{
			vist_col[j]=true;  //放置棋子的列标记
			DFS(i+1,num+1);
			vist_col[j]=false; //回溯后,说明摆好棋子的状态已记录,当前的列标记还原
		}

//	DFS(row+1,num);   //这里是难点,当k<n时,row在等于n之前就可能已经把全部棋子放好
	                  //又由于当全部棋子都放好后的某个棋盘状态已经在前面循环时记录了
	                  //因此为了处理多余行,令当前位置先不放棋子,搜索在下一行放棋子的情况
	return;
}

int main(int i,int j)
{
	while(cin>>n>>k)
	{
		if(n==-1 && k==-1)
			break;

		/*Initial*/

		memset(chess,false,sizeof(chess));
		memset(vist_col,false,sizeof(vist_col));
		status=0;

		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				char temp;
				cin>>temp;
				if(temp=='#')
					chess[i][j]=true;
			}

		DFS(1,0);  //注意初始化的值别弄错了
		cout<<status<<endl;
	}
	return 0;
}

简单模板,一行一行的搜索,和八皇后差不多但是还有一种更快的方法

//Memory Time 
//184K   32MS 
 
#include<iostream>
using namespace std;
 
bool chess[9][9];
bool vist_col[9];  //列标记
int status;  //状态计数器
int n,k;
 
void DFS(int row,int num)  //逐行搜索,row为当前搜索行,num为已填充的棋子数
{
	if(num==k)
	{
		status++;
		return;
	}
 
	if(row>n)    //配合下面DFS(row+1,num); 语句使用,避免搜索越界
		return;
 
	for(int j=1;j<=n;j++)
		if(chess[row][j] && !vist_col[j])
		{
			vist_col[j]=true;  //放置棋子的列标记
			DFS(row+1,num+1);
			vist_col[j]=false; //回溯后,说明摆好棋子的状态已记录,当前的列标记还原
		}
 
	DFS(row+1,num);   //这里是难点,当k<n时,row在等于n之前就可能已经把全部棋子放好
	                  //又由于当全部棋子都放好后的某个棋盘状态已经在前面循环时记录了
	                  //因此为了处理多余行,令当前位置先不放棋子,搜索在下一行放棋子的情况
	return;
}
 
int main(int i,int j)
{
	while(cin>>n>>k)
	{
		if(n==-1 && k==-1)
			break;
 
		/*Initial*/
 
		memset(chess,false,sizeof(chess));
		memset(vist_col,false,sizeof(vist_col));
		status=0;
 
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				char temp;
				cin>>temp;
				if(temp=='#')
					chess[i][j]=true;
			}
 
		DFS(1,0);  //注意初始化的值别弄错了
		cout<<status<<endl;
	}
	return 0;
}

这个代码和本人写的雷同,只是在刚开始的时候标记好了#号,

Time
125ms
Memory
184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
char s[10][10];
int ss[10];
int n,k,num;
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    if(l>n)return ;
        for(int j=0;j<n;j++)
    {
        if(ss[j]||s[l][j]!='#')continue;     //这里有一个妙用
        ss[j]=1;dfs(l+1,step+1);ss[j]=0;
    }
    dfs(l+1,step);
    return ;
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(s,0,sizeof s);
        memset(ss,0,sizeof ss);
        for(int i=0;i<n;i++)
            scanf("%s",&s[i]);
        dfs(0,0);
            cout<<num<<endl;
    }

}

Time
110ms
Memory
184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
char s[10][10];
int ss[10];
int n,k,num;
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    if(l>n)return ;
        for(int j=0;j<n;j++)
    {
        if(!ss[j]&&s[l][j]=='#'){ss[j]=1;dfs(l+1,step+1);ss[j]=0;}    //这里有一个妙用
    }
    dfs(l+1,step);
    return ;
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(s,0,sizeof s);
        memset(ss,0,sizeof ss);
        for(int i=0;i<n;i++)
            scanf("%s",&s[i]);
        dfs(0,0);
            cout<<num<<endl;
    }

}

Time
141ms
Memory
184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
char s[10][10];
int ss[10];
int n,k,num;
bool g[10][10];
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    if(l>n)return ;
    for(int j=0; j<n; j++)
    {
        if(!ss[j]&&g[l][j])
        {
            ss[j]=1;
            dfs(l+1,step+1);
            ss[j]=0;
        }    //这里有一个妙用
    }
    dfs(l+1,step);
    return ;
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(s,0,sizeof s);
        memset(g,false,sizeof g);
        for(int i=0; i<n; i++)
            for(int j=0;j<n;j++)
            {
                char ch;
                cin>>ch;
                if(ch=='#')
                    g[i][j]=true;
            }
        dfs(0,0);
        cout<<num<<endl;
    }

}
Time
110ms
Memory
184kB
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
bool ss[10];
int n,k,num;
bool g[10][10];
void dfs(int l,int step)
{
    if(step==k)
    {
        num++;
        return;
    }
    if(l>n)return ;
    for(int j=0; j<n; j++)
    {
        if(!ss[j]&&g[l][j])
        {
            ss[j]=true;
            dfs(l+1,step+1);
            ss[j]=false;
        }    //这里有一个妙用
    }
    dfs(l+1,step);
    return ;
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        num=0;
        if(n==-1&&k==-1)
            return 0;
        memset(ss,0,sizeof ss);
        memset(g,false,sizeof g);
        for(int i=0; i<n; i++)
            for(int j=0;j<n;j++)
            {
                char ch;
                cin>>ch;
                if(ch=='#')
                    g[i][j]=true;
            }
        dfs(0,0);
        cout<<num<<endl;
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值