網易面試題之 牛牛的作業薄上有一個長度爲 n 的排列 A,這個排列包含了從1到n的n個數,但是因爲一些原因, * 其中有一些位置(不超過 10 個)看不清了,但是牛牛記得這個數列順序對的數量是 k,

package wangyi;

/**
 * Created by Administrator on 2016/12/7.
 * 牛牛的作業薄上有一個長度爲 n 的排列 A,這個排列包含了從1到n的n個數,但是因爲一些原因,
 * 其中有一些位置(不超過 10 個)看不清了,但是牛牛記得這個數列順序對的數量是 k,
 * 順序對是指滿足 i < j 且 A[i] < A[j] 的對數,請幫助牛牛計算出,符合這個要求的合法排列的數目。
 */

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Scanner;

public class Test5{

    /**
     * @param args
     */
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()){
            int RES = 0;
            int n = sc.nextInt();
            int k = sc.nextInt();
            int[] A = new int[n];
            boolean[] flag = new boolean[n+1];
            //flag標記哪些數字已經存在
            for(int i=0;i<n;i++){
                A[i] = sc.nextInt();
                if(A[i] != 0){
                    flag[A[i]] = true;
                }
            }

            //統計排列中不存在的數字
            ArrayList<Integer> list = new ArrayList<Integer>();
            for(int i=1;i<=n;i++){
                if(flag[i] == false)
                    list.add(i);
            }

            //perm用來存模糊數字的全排列
            List<ArrayList<Integer>> perm = new ArrayList<ArrayList<Integer>>();

            //計算perm
            calperm(perm, list, 0);

            //統計已有的排列的順序對
            int cv = 0;
            for(int i=0;i<n;i++){
                if(A[i]!= 0){
                    for(int j=i+1;j<n;j++){
                        if(A[j] != 0 && A[i] < A[j])
                            cv++;
                    }
                }
            }

            //計算每個模糊數字排列的順序對,如果與k相等,則結果RES加一
            for(ArrayList<Integer> tmp : perm){
                int val = cv;
                int[] tmpA = Arrays.copyOf(A, n);
                val += calvalue(tmp, tmpA);
                if(val == k)
                    RES++;
            }

            System.out.println(RES);
        }
    }

    //計算排列的順序對
    public static int calvalue(List<Integer> list, int[] A){
        int val = 0;
        int j = 0;
        for(int i=0;i<A.length;i++){
            if(A[i] == 0){
                A[i] = list.get(j++);
                for(int k = 0;k<i;k++){
                    if(A[k]!=0 && A[k]<A[i])
                        val++;
                }
                for(int k=i+1;k<A.length;k++){
                    if(A[k]!=0 && A[k]>A[i])
                        val++;
                }
            }
        }
        return val;
    }

    //計算全排列
    public static void calperm(List<ArrayList<Integer>> perm , ArrayList<Integer> list, int n){
        if(n == list.size()){
            perm.add(new ArrayList<Integer>(list));
        }else{
            for(int i=n;i<list.size();i++){
                Collections.swap(list, i, n);
                calperm(perm, list, n+1);
                Collections.swap(list, i, n);
            }
        }
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章