贪心算法-移除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;
	}
}
发布了47 篇原创文章 · 获赞 7 · 访问量 5879

猜你喜欢

转载自blog.csdn.net/weixin_42227576/article/details/98482759
今日推荐