小米oj 9 移除 K 位得到最小值

題目:https://code.mi.com/problem/list/view?id=9
AC代碼(python3)如下:

import sys

for line in sys.stdin:
    line = line.strip()
    num,K = line.split(' ')
    K = int(K)

    for i in range(K):
        for j in range(9,-1,-1):
            if str(j) in num:
                t=num.index(str(j))
                #剩餘需要去掉的個數K-i,至少t前面的都去掉,才能滿足
                #如果K-i<t,也就是此步輪不到t位置元素去掉,而應該去掉更高位數字
                #純屬意外想到的,啊哈!
                if K-i>=t:
                    num = num[:t]+num[t+1:]
                    break

    print(int(num))

思路:每次找最佳的數位,並去掉該數位。註釋部分已有說明,此處的t,雖然是索引值對應的位置,但在K-i>=t,應理解爲去掉的num[t],實際上是第t+1位元素,它前面元素(高位數字)的個數爲t。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章