【Lintcode】188. Insert five

题目地址：

https://www.lintcode.com/problem/insert-five/description

给定一个数$x$，且$|x|<10^6$，要求在这个数的某个位上插入一个$5$使得所得新数最大。

如果$x=0$显然答案是$50$。接下来需要按照$x$是正还是负进行讨论。如果$x$为正，那么只需要从左向右看$x$的每一位，当发现第一个小于等于$5$的数的时候，直接将$5$插在这里即可；如果$x$为负，那么只需要从左向右看$-x$的每一位，当发现第一个大于等于$5$的数的时候，直接将$5$插在这里即可。如果遍历完$x$的每一位都没发现大于等于（或者当$x<0$，小于等于）$5$的数，则将$5$加在末尾。具体证明可以用分类讨论的办法，只需证明插在别的位置所得答案都不如上述算法的位置大即可。代码如下：

public class Solution {
    /**
     * @param a: A number
     * @return: Returns the maximum number after insertion
     */
    public int InsertFive(int a) {
        // write your code here
        // 先把a的绝对值求出来，然后转为StringBuilder
        int b = Math.abs(a);
        StringBuilder sb = new StringBuilder(String.valueOf(b));
        // 取出a的位数，后面有用
        int len = sb.length();
        
        if (a >= 0) {
            for (int i = 0; i < len; i++) {
                if (sb.charAt(i) <= '5') {
                    sb.insert(i, '5');
                    break;
                }
            }
        } else {
            for (int i = 0; i < len; i++) {
                if (sb.charAt(i) >= '5') {
                    sb.insert(i, '5');
                    break;
                }
            }
        }
        // 这里如果发现sb的长度没有变化，说明5没有被插入进去，则append到末尾
        if (sb.length() == len) {
            sb.append(5);
        }
        
        int res = Integer.parseInt(sb.toString());
        return a >= 0 ? res : -res;
    }
}

时空复杂度$O(\log a)$。