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;
    }

}

猜你喜欢

转载自blog.csdn.net/weixin_51626694/article/details/116101269