华为机试-移除K位数字

华为机试-移除K位数字_第1张图片

1.思路

这题没想到好思路,使用了暴力dfs算法,尽量使用剪枝函数优化

2.代码

public class Main{
//初始化结果为999999999,方便后面和运算的结果比较,取最小值
 private static String ans = "999999999";
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s = sc.nextLine();
        int r = sc.nextInt();
        if(s == "") return;
        for(int i =0;i<s.length() - r;i++){
            dfs(1,i,s,s.charAt(i)+"",s.length() - r);
        }
        System.out.println(ans);
    }

    public static void dfs(int k,int index,String s,String t,int n){
    	//结束条件,选取了k个数了,如果比之前的结果小,就保留并return。比之前的结果大直接return
        if(k == n){
            ans = t.compareTo(ans) < 0 ? t : ans;
            return ;
        }
        //剪枝,如果当前选取的数字字符串,已经比先前计算的字符串要大,直接return
        if(ans.compareTo(t) < 0) return;
        for(int i = index + 1;i<s.length();i++){
            dfs(k+1,i,s,t+s.charAt(i),n);
        }
    }

}

你可能感兴趣的:(算法,华为机试,深度优先,算法)