有重复元素的排列问题(Java)

Posted 2021-11-04 nuist__NJUPT

思想：用前缀法求全排列，用set集合去除重复即可。

import java.util.Iterator;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Permutation {
    private static Set<String> set = new TreeSet<>(); //set集合除去重复
    public static boolean check(String prefix, char [] arr, char c){//c在前缀中的个数要小于在数组中的个数才能递归
        int sum1 = 0, sum2 = 0 ;
        for(int i=0; i<prefix.length(); i++){
            if(c == prefix.charAt(i)){
                sum1 ++ ;
            }
        }
        for(int j=0; j<arr.length; j++){
            if(c == arr[j]){
                sum2 ++ ;
            }
        }
        if(sum2 > sum1){
            return true ;
        }else{
            return false ;
        }
    }
    public static void permutation(String prefix, char [] arr){ //求全排列
        if(prefix.length() == arr.length){
            set.add(prefix) ;
            return ;
        }
        for(int i=0; i<arr.length; i++){
            char c = arr[i] ;
            if(check(prefix, arr, c))
            permutation(prefix+c, arr);
        }
    }
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in) ;
        int n = input.nextInt() ;
        char [] c = new char [n] ;
        String s = input.next() ;
        for(int i=0; i<n; i++){
            c[i] = s.charAt(i) ;
        }
        permutation("", c);
        Iterator iterator = set.iterator() ;
        while(iterator.hasNext()){
            System.out.println(iterator.next()) ;
        }
    }
}