LintCode 570. 寻找丢失的数 II

Posted 2020-10-30 J1ac

给一个由 1 - n 的整数随机组成的一个字符串序列，其中丢失了一个整数，请找到它。

 注意事项

n <= 30

样例

给出 n = 20, str = 19201234567891011121314151618

丢失的数是 17 ，返回这个数。

思路：回溯法进行深度优先搜索，需要注意的几点：数字的越界检查，找到的数字的计数（不计数可能碰到一些特殊情况：比如最终缺失存在两个数，而其中一个数其实可以分拆成两个数）不可缺少

 

class Solution {
public:
    /**
     * @param n: An integer
     * @param str: a string with number from 1-n in random order and miss one number
     * @return: An integer
     */
    int findMissing2(int n, string &str) {
        // write your code here
        vector<int> dict(n+1,0);	//数组用于标记数字是否出现过，dict[0]用来记录已经出现了多少个数字
        return findTheNum(dict,0,str);
    }

    int findTheNum(vector<int>& d,int index,string & str){
        int res = -1;
        if(index == str.size() && d[0] == d.size()-2){	//迭代需要的出口，当index到达str最后的下一个位置并且数字总数达到n-1个时返回剩余的数
            for(int i = 1; i < d.size(); ++i){
                if(d[i] == 0) res = i;
            }
        }
        if(str[index] == ‘0‘) return res; //指向的下一个数以0开始直接返回
        else{
            int temp1 = str[index] - ‘0‘;	//第一个数字
            if(d[temp1] == 0) {
                d[temp1] = 1; //深度优先搜索，找合适的结果
                d[0]++;	//计数+1
                int ret = findTheNum(d,index+1,str);
                if(ret != -1) res = ret; 	//如果能找到更改res值
                d[temp1] = 0;//回溯的套路，恢复原状
                d[0]--;
            }
            if(index+1<str.size()){
                int temp2 = str[index+1] - ‘0‘;
                temp2 = temp2 + temp1*10; //如果两个数组成
                if(temp2 < d.size() && d[temp2] == 0){
                    d[temp2] = 1;
                    d[0]++;
                    int ret = findTheNum(d,index+2,str);
                    if(ret != -1) res = ret;
                    d[temp2] = 0;
                    d[0]--;
                }
            }
        }
        return res; //最后返回结果
    }

};

　　其实只要熟悉了回溯的套路，这种题只要认真审题做出来就是时间问题了