網易面試題之 牛牛的作業薄上有一個長度為 n 的排列 A,這個排列包含了從1到n的n個數,但是因為一些原因, * 其中有一些位置(不超過 10 個)看不清了,但是牛牛記得這個數列順序對的數量是 k,
阿新 • • 發佈:2019-01-03
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); } } } }