[LintCode 570.] 寻找丢失的数 II

LintCode 570. 寻找丢失的数 II

问题描述

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

样例

样例1

输入: n = 20 和 str = 19201234567891011121314151618
输出: 17
解释:
19'20'1'2'3'4'5'6'7'8'9'10'11'12'13'14'15'16'18
样例2

输入: n = 6 和 str = 56412
输出: 3
解释:
5'6'4'1'2

注意事项

n < 100
数据保证有且仅有唯一解
如果您找到的答案列表丢失了多个数字，可能是您没有找到正确的字符串拆分方式

解题思路

尝试拆分字符串成数字，如果能恰好拆分成 n-1 个不同的数，并且都在 1-n的范围内，那么剩下的一个恰好就是所得。
如果拆分的结果不合适，继续尝试下一种拆分方式。
使用一个bool数字标记拆分出的数字，用于去重；
使用一个变量标记已经拆分出的数字，用于计数；
当已经拆分出n-1个数，并且字符用完的时候，剩下的数就是结果。

参考代码

class Solution {
    int missing;
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
        missing = -1;
        vector<bool> vis(n+1, false);
        dfs(str, n, vis, 0);
        return missing;
    }
    void dfs(string str, int n, vector<bool>& vis, int count) {
        if (str.length() == 0 && count == n-1) {
            for (int i=1; i<=n; i++) {
                if (!vis[i]) missing = i;
            }
        }
        for (int i = 1; i <= 2 && i <= str.length(); i++) {
            int num = stoi(str.substr(0, i));
            // cout << num << ' ';
            if (num == 0 || num > n || vis[num]) {
                continue;
            }
            vis[num] = true;
            dfs(str.substr(i), n, vis, count + 1);
            vis[num] = false;
        }
    }
};