力扣564——寻找最近的回文数

解题思路

分情况讨论：突破点在于由一个回文串得到其下一个和上一个回文串；

如果初始不是回文数，先按前半部分反转构造回文数（可以反证法证明这是最接近的之一，如果比初始大，就是大的里面最接近的，如果比初始小，就是小的里面最接近的），如果构造的比原来大，那求构造的上一个回文数，否则求下一个回文数，二者比较求答案；

如果初始就是回文数，那么直接求该数的上一个和下一个回文数，比较得答案；

对于一个回文数，求其上一个和下一个回文数，就是按中间位往两边减或加，注意0和9的特殊处理，需要不断进位；

另外对于特殊边界情况，如99的下一个（101），可以通过加二得到；1001的上一个（999），可以减二得到。

另外数据范围1e18，刚好够long long；

代码

class Solution {
public:
    bool ispal(string s) {
        int len = s.size();
        for(int i = 0; i < len / 2; i++) {
            if(s[i] != s[len - 1 - i]) return false;
        }
        return true;
    }
    string next(string s) {
        int n = s.size(); int pos = n / 2;
        string res = s;
        if(n % 2) {
            if(res[pos] != '9') {
                res[pos]++; return res;
            }
        }
        else {
            if(res[pos] != '9') {
                res[pos]++;
                res[pos - 1]++;
                return res;
            }
            pos--;
        }
        while(pos >= 0 && res[pos] == '9') {
            res[pos] = '0';
            pos--;
        }
        if(pos < 0) return to_string(stoll(s) + 2);
        res[pos]++;
        for(int i = 0; i < n / 2; i++) res[n - 1 - i] = res[i];
        return res;
    }
    string last(string s) {
        int n = s.size(); int pos = n / 2;
        string res = s;
        if(n % 2) {
            if(res[pos] != '0') {
                res[pos]--; return res;
            }
        }
        else {
            if(res[pos] != '0') {
                res[pos]--;
                res[pos - 1]--;
                if(res[0] == '0') return to_string(stoll(s) - 2);
                return res;
            }
            pos--;
        }
        while(res[pos] == '0') {
            res[pos] = '9';
            pos--;
        }
        res[pos]--;
        if(res[0] == '0') return to_string(stoll(s) - 2);
        for(int i = 0; i < n / 2; i++) res[n - 1 - i] = res[i];
        return res;
    }
    string nearestPalindromic(string n) {
        string ans = n;
        int len = n.size();
        if(len == 1) return to_string(stoi(n) - 1);
        int pos = len / 2;
        if(!ispal(n)) {
            while(n[pos] == n[len - 1 - pos]) pos--;
            for(int i = 0; i <= pos; i++) ans[len - 1 - i] = ans[i];
            long long ll_n = stoll(n), ll_ans = stoll(ans);
            if(stoll(ans) > stoll(n)) {
                string tmp = last(ans);
                long long ll_tmp = stoll(tmp);
                return ll_ans - ll_n >= ll_n - ll_tmp ? tmp : ans;
            }
            else {
                string tmp = next(ans);
                long long ll_tmp = stoll(tmp);
                return ll_n - ll_ans > ll_tmp - ll_n ? tmp : ans;
            }
        }
        else {
            string nextpal = next(n), lastpal = last(n);
            long long ll_n = stoll(n), ll_next = stoll(nextpal), ll_last = stoll(lastpal);
            return ll_n - ll_last > ll_next - ll_n ? nextpal : lastpal;
        }
    }
};