蓝桥杯2015年第六届真题-机器人塔 （C++代码）

解题思路:位向量法

注意事项:时间超限，显然这里的数据测试不是很严格，数据直接通过了

参考代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int a[50],M,N,sum,x;

void dfs(int cur)
{
	if(cur==x)
	{
		int z=x,numa=0,numb=0;
		while(z)
		{
			for(int i=0;i<z;i++)
				if(a[i]) numa++;
				else numb++;
			z--;
			for(int i=0;i<z;i++)
				if(a[i]==a[i+1]) a[i]=1;
				else a[i]=0;
		}
		if(numa==M && numb==N) sum++;
	}
	else 
	{
		a[cur]=1;
		dfs(cur+1);
		a[cur]=0;
		dfs(cur+1); 
	}
}

int main(void)
{
	cin >> M >> N;
	int i=1;
	while(i*(i+1)/2 != M+N) i++;
	x = i;
	dfs(0);
	cout << sum << endl;
	return 0;
}