程序员面试金典 10.5

Sparse Search:在一个有序的字符串数组中查找指定字符串,字符串数组中可能存在空字符串。

如果没有空字符串,二分查找就可以了。现在有空字符串,依然可以借用二分查找的思路,如果mid不行落在了一个空字符串上,就将其移动到最近的一个非空字符串进行后面的查找。

class Solution {
public:
    int findString(vector<string>& words, string s) {
        return findString(words, s, 0, words.size());
    }
private:
    int findString(const vector<string> &words, const string &s, int begin, int end)
    {
        if(begin >= end) return -1;
        int mid = begin + (end - begin) / 2;
        mid = shiftMid(words, mid, begin, end);
        if(mid == -1) return -1;
        if(words[mid] == s) return mid;
        if(s < words[mid]) return findString(words, s, begin, mid);
        else return findString(words, s, mid + 1, end);
    }
    int shiftMid(const vector<string> &words, int mid, int begin, int end)
    {
        int left = mid, right = mid;
        while(left >= begin && right < end){
            if(!words[left].empty()) return left;
            else if(!words[right].empty()) return right;
            left--, right++;
        }
        return -1;
    }
};
发布了194 篇原创文章 · 获赞 2 · 访问量 7704

猜你喜欢

转载自blog.csdn.net/RayoNicks/article/details/105532507