NYOJ491. 幸运三角形（DFS）

幸运三角形

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 3

描述

        话说有这么一个图形，只有两种符号组成（‘+’或者‘-’），图形的最上层有n个符号，往下个数依次减一，形成倒置的金字塔形状，除第一层外（第一层为所有可能情况），每层形状都由上层决定，相邻的符号相同，则下层的符号为‘+’，反之，为‘-’;如下图所示（n = 3 时的两种情况)：

                                           

如果图中的两种符号个数相同，那这个三角形就是幸运三角形，如上图中的图（2）.

输入

有多组测试数据（少于20组）。
每行含一个整数n（0<n<20)。

输出

输出相应的幸运三角形个数。

样例输入

3
4

样例输出

4
6

【分析】DFS

法一：朴素解法，TLE

#include <stdio.h>
#include <string.h>
#define maxn 25
int N,ans,sum;
int a[maxn][maxn];
int vis[maxn];
void dfs(int cur,int a0,int a1)
{
	int i,j;
	int ans0,ans1;
	
	if(cur==N)
	{
		ans0=a0,ans1=a1;
		for(i=1;i<cur;i++)
		{
			for(j=0;j<cur-i;j++)
			{
				if(ans0>temp/2 || ans1>temp/2)
					break;
				if(a[i-1][j]==a[i-1][j+1])
				{
					a[i][j]=1;
					ans1++;
				}
				else
				{
					a[i][j]=0;
					ans0++; 
				} 
			}
		}
		if(ans0==ans1)
			ans++;
		return;
	}
	for(i=0;i<=1;i++)
	{
		a[0][cur]=i;
		if(i==0)
			dfs(cur+1,a0+1,a1);
		else
			dfs(cur+1,a0,a1+1);
		a[0][cur]=0;
	}
}
int main()
{
	while(scanf("%d",&N)!=EOF)
	{
		sum=(N*(N+1))/2;
		if(sum%2!=0)
		{
			printf("0\n");
			continue;
		}
		ans=0;
		memset(vis,0,sizeof(vis));
		dfs(0,0,0);
		printf("%d\n",ans);
	}
	return 0;
}

法二：DFS+回溯

#include <stdio.h>
#include <string.h>
#define maxn 25
int N,ans;
int cnt,sum;  //cnt-'-'号个数 
int a[maxn][maxn];
int vis[maxn];
void dfs(int cur)
{
	int i,j;
	int ans0,ans1;
	if(cur>N)
		ans++;
	else
	{
		//0-'+' 1-'-'
		for(i=0;i<=1;i++)
		{
			a[1][cur]=i;
			cnt+=i;
			for(j=2;j<=cur;j++)
			{
				a[j][cur-j+1]=a[j-1][cur-j+1]^a[j-1][cur-j+2];
				cnt+=a[j][cur-j+1];
			}
			if((cnt<=sum/2) && (cur*(cur+1)/2-cnt<=sum/2))
				dfs(cur+1);
			for(j=2;j<=cur;j++)
				cnt-=a[j][cur-j+1];
			cnt-=i;
		} 
	} 
}
int main()
{
	while(scanf("%d",&N)!=EOF)
	{
		sum=(N*(N+1))/2;
		if(sum%2!=0)
		{
			printf("0\n");
			continue;
		}
		ans=0;
		memset(vis,0,sizeof(vis));
		dfs(1);
		printf("%d\n",ans);
	}
	return 0;
}