2019年腾讯提前批笔试编程题第五题

2019年腾讯提前批笔试编程题第五题

题目描述

小Q给你n个数1、2、3…n， 代表n个楼，第i个楼的高度为i，每个楼会有一种颜色现在问有多少的排列满足从左往右（站在左边很远的地方看）看能看到L种颜色（即看到了L-1次颜色的变化），答案对1e9+9取模。
如果两个相同颜色楼的高度分别问H1，H2（H1 你能看到一个楼的前提是这个楼之前的楼都比它矮

输入描述：

第一行输入两个整数n，L(1 ≤ n ≤ 1296)，（1 ≤ L ≤ 1296），（1≤L≤n）
第二行输入n个整数，c_i表示每个楼的颜色（1≤c_i≤n）

输出描述：

输出一个整数

示例

输入：

4 3
1 1 2 1 

输出：

6

import java.util.*;

/**
 * @author Milingyun
 * @date 2019-03-09 19:36
 */
public class Building {
	//全排列
    public static List<List<Integer>> permute(int[] nums)
    {
        List<List<Integer>> all=new ArrayList<List<Integer>>();
        allSort(nums, 0, nums.length-1, all);
        return all;
    }
    public static void allSort(int[] array,int begin,int end,List<List<Integer>> all)
    {
        if(begin==end){
            List<Integer> origi=new ArrayList<Integer>();
            for(int a:array)
            {
                origi.add(a);
            }
            all.add(origi);
            return;
        }

        for(int i=begin;i<=end;i++){
            swap(array,begin,i );
            allSort(array, begin+1, end,all);
            swap(array,begin,i );
        }

    }
    public static void swap(int[] array,int a,int b){
        int tem=array[a];
        array[a]=array[b];
        array[b]=tem;
    }

    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        String[] input=sc.nextLine().split(" ");
        int n=Integer.parseInt(input[0]);
        int l=Integer.parseInt(input[1]);
        String[] lou=sc.nextLine().split(" ");
        HashMap builds=new HashMap();
        for(int i=0;i<n;i++){
            builds.put(i+1,Integer.parseInt(lou[i]));
        }
        int[] nums=new int[n];
        for(int i=0;i<n;i++){
            nums[i]=i+1;
        }
       //所有的排列方式
       List pailie=permute(nums);
        int result=0;
        for(int i=0;i<pailie.size();i++){
            List<Integer> target= (List) pailie.get(i);
            int max=target.get(0);
            int pre= (int) builds.get(max);
            int count=1;
            for(int j=1;j<target.size();j++){
                int tall=target.get(j);
                int color= (int) builds.get(tall);
                if(tall<=max){continue;}
                max=tall;
                if(color==pre){ continue;}
                count++;
                pre=color;

            }

            if(count==l){
                result++;
            }
        }
        System.out.println(result);
    }

}

代码测试通过95%，5%应该是由于答案没有对1e9+9取模。