天池 在线编程 插入五

样例  1:
输入:  a = 234
输出: 5234    

class Solution {
public:
    /**
     * @param a: A number
     * @return: Returns the maximum number after insertion
     */
    int InsertFive(int a) {
        // write your code here
        bool negative = a < 0;
        a = abs(a);
        string num = to_string(a);
        int ans = 0;
        if(!negative)
        {
            bool insert = false;
            for(int i = 0; i < num.size(); i++)
            {	// 非负数，找到第一个小于 5 的数字
                if(num[i] < '5')
                {
                    num.insert(num.begin()+i, '5');
                    insert = true;
                    break;
                }
            }
            if(!insert)//没有插入过，放在末尾
                num += '5';
        }
        else
        {
            bool insert = false;
            for(int i = 0; i < num.size(); i++)
            {	// 负数，找到第一个大于 5 的数字
                if(num[i] > '5')
                {
                    num.insert(num.begin()+i, '5');
                    insert = true;
                    break;
                }
            }
            if(!insert)//没有插入过，放在末尾
                num += '5';
        }
        for(int i = 0; i < num.size(); ++i)
            ans = ans*10+num[i]-'0';//转成int
        if(negative)
            return -ans;
        return ans;
    }
};