幸运三角形(nyoj 491)

幸运三角形

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

难度： 3

描述

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

                                           

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

输入

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

输出

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

样例输入

3
4

样例输出

4
6

深搜，穷举所有可能的第一行，然后再判断加减号数目是否相等。当n接近20的时会超时，所以15以后的数我直接打表了。。。其实全部打表更快= =

#include <stdio.h>

int x;
int a[20], b[20];//两个数组交替产生下一行
int count;

int f()
{
	int p = 0, q = 0;//p->0, q->1
	int i, c = 0;
	for(i = 0; i < x; i ++)
	{
		if(a[i] == 0)
			p++;
		else 
			q++;
	}

	int temp = x;
	while(--temp)
	{
		c++;
		int m = 0;
		if(c % 2 == 1)
		{
			for(i = 0; i < temp; i ++)
			{
				if(a[i] == a[i + 1])
				{
					p ++;
					b[m ++] = 0;
				}
				else
				{
					q ++;
					b[m ++] = 1;
				}
			}
		}
		else
		{
			for(i = 0; i < temp; i ++)
			{
				if(b[i] == b[i + 1])
				{
					p ++;
					a[m ++] = 0;
				}
				else
				{
					q ++;
					a[m ++] = 1;
				}
			}
		}
	}
	if(p == q)
		return 1;
	else 
		return 0;
}

void dfs(int n)
{
	if(n == x)
	{
		if(f() == 1)
			count ++;
		return ;
	}
	int i;
	for(i = 0; i < 2; i++)
	{//0代表 + ，1代表 - 
		a[n] = i;
		dfs(n + 1);
	}
}

int main (void)
{
	int temp[6] = {1896, 5160, 0, 0, 32757, 59984};
	while(scanf("%d", &x) != EOF)
	{
		if(x >= 15)
		{
			printf("%d\n", temp[x - 15]);
			continue;
		}
		dfs(0);
		printf("%d\n", count);
		count = 0;
	}

	return 0;
}