搭积木【蓝桥杯】

小明最近喜欢搭数字积木，
一共有10块积木，每个积木上有一个数字，0~9。

搭积木规则：
每个积木放到其它两个积木的上面，并且一定比下面的两个积木数字小。
最后搭成4层的金字塔形，必须用完所有的积木。

下面是两种合格的搭法：

0
1 2
3 4 5
6 7 8 9

0
3 1
7 5 2
9 8 6 4

请你计算这样的搭法一共有多少种？

请填表示总数目的数字。
注意：你提交的应该是一个整数，不要填写任何多余的内容或说明性文字。
————————————————

考点：全排列+判断+剪枝

get:在做全排列的题最好输出一个每个符合条件的结果方便肉眼检查是否正确无误！

全排列代码：

private static void f(int[] a, int k){
        if (a[0] != 0){
            return;
        }

        if (check(a) && k == a.length){//此处应该等全排列完后再进行判断，否则会有重复统计
            System.out.println("   "+a[0]);
            System.out.println("  "+a[1]+" "+a[2]);
            System.out.println(" "+a[3]+" "+a[4]+" "+a[5]);
            System.out.println(a[6]+" "+a[7]+" "+a[8]+" "+a[9]);
            ans++;
        }

        for ( int i = k; i < a.length; i++ ){

            int t = a[i];
            a[i] = a[k];
            a[k] = t;

            f(a,k+1);

            t = a[i];
            a[i] = a[k];
            a[k] = t;

        }
    }

判断代码：

 private static boolean check(int[] a) {

        boolean flag = true;

        if (a[0] > a[1] || a[0] > a[2]) flag = false;
        if (a[1] > a[3] || a[1] > a[4]) flag = false;
        if (a[2] > a[4] || a[2] > a[5]) flag = false;
        if (a[3] > a[6] || a[3] > a[7]) flag = false;
        if (a[4] > a[7] || a[4] > a[8]) flag = false;
        if (a[5] > a[8] || a[5] > a[9]) flag = false;

        return flag;
    }

全部代码如下：

package pastExamPaper;
/*
    搭积木(全排列+判断)  全排列可以通过剪枝来提高效率，同时要注意在判断满足题意的同时应该满足k==a.length这个条件
    收获：在做全排列的题最好输出一个每个符合条件的结果方便肉眼检查是否正确无误！
 */
public class Demo58 {

    private static int ans;

    public static void main(String[] args) {


        int[] a = {0,1,2,3,4,5,6,7,8,9};
        f(a,0);
        System.out.println(ans);//第一次计算答案为1353，正确答案为768

    }

    private static void f(int[] a, int k){
        if (a[0] != 0){
            return;
        }

        if (check(a) && k == a.length){//此处应该等全排列完后再进行判断，否则会有重复统计
            System.out.println("   "+a[0]);
            System.out.println("  "+a[1]+" "+a[2]);
            System.out.println(" "+a[3]+" "+a[4]+" "+a[5]);
            System.out.println(a[6]+" "+a[7]+" "+a[8]+" "+a[9]);
            ans++;
        }

        for ( int i = k; i < a.length; i++ ){

            int t = a[i];
            a[i] = a[k];
            a[k] = t;

            f(a,k+1);

            t = a[i];
            a[i] = a[k];
            a[k] = t;

        }
    }

    private static boolean check(int[] a) {

        boolean flag = true;

        if (a[0] > a[1] || a[0] > a[2]) flag = false;
        if (a[1] > a[3] || a[1] > a[4]) flag = false;
        if (a[2] > a[4] || a[2] > a[5]) flag = false;
        if (a[3] > a[6] || a[3] > a[7]) flag = false;
        if (a[4] > a[7] || a[4] > a[8]) flag = false;
        if (a[5] > a[8] || a[5] > a[9]) flag = false;

        return flag;
    }

}