机器人塔

机器人塔


X星球的机器人表演拉拉队有两种服装，A和B。
他们这次表演的是搭机器人塔。


类似：


     A
    B B
   A B A
  A A B B
 B B B A B
A B A B B A


队内的组塔规则是：
  
  A 只能站在 AA 或 BB 的肩上。
  B 只能站在 AB 或 BA 的肩上。


你的任务是帮助拉拉队计算一下，在给定A与B的人数时，可以组成多少种花样的塔。


输入一行两个整数 M 和 N，空格分开（0<M,N<500）,分别表示A、B的人数，保证人数合理性。


要求输出一个整数，表示可以产生的花样种数。


例如：
用户输入：
1 2


程序应该输出：
3




再例如：
用户输入：
3 3


程序应该输出：
4


资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms




请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。


所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。

注意：主类的名字必须是：Main，否则按无效代码处理。

写了两种算法，第一种超限，现在给出第一种代码

package cn.lml.Day5.examJavaA;
/*
 
机器人塔


X星球的机器人表演拉拉队有两种服装，A和B。
他们这次表演的是搭机器人塔。


类似：


     A
    B B
   A B A
  A A B B
 B B B A B
A B A B B A


队内的组塔规则是：
  
  A 只能站在 AA 或 BB 的肩上。
  B 只能站在 AB 或 BA 的肩上。


你的任务是帮助拉拉队计算一下，在给定A与B的人数时，可以组成多少种花样的塔。


输入一行两个整数 M 和 N，空格分开（0<M,N<500）,分别表示A、B的人数，保证人数合理性。


要求输出一个整数，表示可以产生的花样种数。


例如：
用户输入：
1 2


程序应该输出：
3




再例如：
用户输入：
3 3


程序应该输出：
4


资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms




请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。


所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意：主类的名字必须是：Main，否则按无效代码处理。


//采用dfs暴力破解，但是时间复杂度比较高



 */
import java.util.Scanner;


public class num4 {
static Scanner s=new Scanner(System.in);
static int A=s.nextInt();
static int B=s.nextInt();
static int a[]=new int [A+B+1];
static int map[][]=null;
static int count=0;
static int max=A>B?A:B;
public static void main(String[] args) {
// TODO Auto-generated method stub

map=new int[max+1][max+1];
for(int i=0;i<A;i++){
a[i]=0;//如果是A就存0
}
for(int i=A;i<A+B;i++){
a[i]=1;//如果是B就存1
}
//System.out.println(max);
//check(a);
prems(a,0);
System.out.println(count);
}
public static void prems(int a[],int start){
//交换大法排重复的数组
if(start==(A+B-1)){
// /System.out.println(111111);
check(a);
return;
}
for(int i=start;i<A+B;i++){
//System.out.println(111111+"*******");
if(ok(a,start,i)){

int temp=a[start];
a[start]=a[i];
a[i]=temp;
prems(a,start+1);
int temps=a[start];
a[start]=a[i];
a[i]=temps;
}
}
}
public static boolean ok(int a[],int begin,int end){
//start和i之间是否有重复
for(int k=begin;k<end;k++){
if(a[k]==a[end]){
return false;
}
}
return true;
}
public static void check(int b[]){
//将一维数组转为二维数组，并对max-1层的下面一个进行判断，有一个不满足条件就把
//ok变为false，如果全部满足条件，则最后count++
int index=0;
out:for(int i=0;i<max;i++){
for(int j=0;j<=i;j++)
{
map[i][j]=a[index++];
if(index==A+B){
break out;
}
// System.out.print(map[i][j]+" ");
}
//System.out.println();

}
boolean ok=true;
for(int i=0;i<max-1;i++)
{
for(int j=0;j<=i;j++){
if(map[i][j]==0){
if(map[i+1][j]!=map[i+1][j+1]){
ok=false;
}
}
if(map[i][j]==1){
if(map[i+1][j]==map[i+1][j+1]){
ok=false;
}
}
}
//System.out.println();
}
if(ok==true){
count++;
}
}
}