NYOJ491. 幸运三角形(DFS)

19 篇文章 0 订阅

幸运三角形

时间限制: 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;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值