貪心算法-移除K個數字(2018小米秋招 筆試)

題目描述:有一十進制正整數,移除其中的 K 個數,使剩下的數字是所有可能中最大的。
假設:
字符串的長度一定大於等於 K
字符串不會以 0 開頭

輸入:1432219
           3

輸出:4329

算法:

import java.util.Scanner;
import java.util.Stack;

public class DeleteK {
	public static void main(String[] args) {
		Scanner scanner=new Scanner(System.in);
		String num=scanner.nextLine();
		int k=scanner.nextInt();
		String resulteString=DeleteK.removeK(num, k);
		System.out.println(resulteString);
	}
	
	public   static String removeK(String num,int k) {
		String result="";
		//根據數值比較進行入棧移除
		Stack<Integer> stack=new Stack<Integer>();
		for (int i = 0; i < num.length(); i++) {
			int temp=num.charAt(i)-'0';
			//進行比較操作
			while (stack.size()!=0&&stack.peek()<temp&&k>0) {
				stack.pop();
				k--;
			}
			//比較完成後,執行入棧操作
			if (stack.size()!=0||temp!=0) {
				stack.push(temp);
			}
		}
			//如果存在數據全部存入棧中,但是k的值不爲0的
			while (k>0&&stack.size()!=0) {
				stack.pop();
				k--;
			}
		
			//將數據從棧中全部取出
			while (stack.size()!=0) {
				result=stack.pop()+result;
			}
		
		return result;
	}
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章