触宝线上笔试2018

题目：有多个case，每个case中有一个案例库，求给定字符串是否是案例库中的子串，若是输出个数。

输入：第一行，输出案例库中字符串的个数N，下面N行数为每一个字符串；给定字符串的个数为M，下面M行为给定字符串

例子：

3           //case1案例库
aaa
aaa
baa
2           //给定字符串
aa
ba
1           //case2 案例库
a 
1            //给定字符串
a

输出
3       //aa是案例库1中三个字符串的子串，个数为3
1       //ba是案例库1中第三个字符串的子串，故个数为1
1       //a是案例库2中的字串，个数为1

解法：用给定的字符串轮流的和案例库中字符比较，看是否是其子串，若是，则个数加1。

#include<iostream>
#include<string>
#include<vector>

using namespace std;

bool isMatch(const string &dict, const string &example);
vector<int> getNext(const string &needle, int m, vector<int> next);

int main()
{
    while (cin)        //此处是否正确？
    {
        int N;
        cin >> N;
        vector<string> dict(N, "");
        for (int i = 0;i<N;i++)
        {
            cin >> dict[i];
        }

        int M;
        cin >> M;
        vector<string> example(M, "");
        for (int i = 0;i<M;i++)
        {
            cin >> example[i];
        }

        for (int i = 0;i<M;i++)
        {
            int count = 0;
            for (int j = 0;j<N;j++)
            {
                if (isMatch(dict[j], example[i]))
                {
                    count++;
                }
            }
            cout << count << endl;        //输出
        }
    }
    return 0;
}

//KMP匹配算法
bool isMatch(const string &dict, const string &example)
{
    int n = dict.size(), i = 0;
    int m = example.size(), j = 0;

    if (m>n) return false;
    if (n == 0 || m == 0)  return true;

    vector<int> next(m);
    next = getNext(example, m, next);

    while (j<m&&i<n)
    {
        if ((0>j) || dict[i] == example[j])
        {
            i++;
            j++;
        }
        else
        {
            j = next[j];
        }
    }
    return j == m ? true : false;
}

vector<int> getNext(const string &needle, int m, vector<int> next)
{
    int t = -1;
    next[0] = -1;
    int j = 0;
    while (j<m - 1)
    {
        if (0>t || needle[j] == needle[t])
        {
            j++;
            t++;
            next[j] = (needle[j] != needle[t] ? t : next[t]);
        }
        else
            t = next[t];
    }
    return next;
}

做题失败的原因，主要在对多个case之间的交替处理上，还有对输出时考虑失误，不一定要先保存输出结果，然后再输出。好几次笔试都是在输入输出上存在问题，在这方面要引起警觉。