力扣练习——70 串联所有单词的子串

1.问题描述

给定一个字符串 s 和一些长度相同的单词 words。找出 s 中恰好可以由 words 中所有单词串联形成的子串的起始位置。

注意子串要与 words 中的单词完全匹配，中间不能有其他字符，但不需要考虑 words 中单词串联的顺序。

示例 1：

输入：

  s = "barfoothefoobarman",

  words = ["foo","bar"]

输出：0 9

解释：

从索引 0 和 9 开始的子串分别是 "barfoo" 和 "foobar" 。输出时，按照索引由小到大顺序输出。

 

示例 2：

输入：

  s = "wordgoodgoodgoodbestword",

  words = ["word","good","best","word"]

输出：-1

s中的子串无法由words串联得到，所以输出-1

 

可使用以下main函数：

int main()

{

    string s,str;

    vector<string> words;

    int n;

    cin>>s;

    cin>>n;

    for(int i=0; i<n; i++)

    {

        cin>>str;

        words.push_back(str);

    }

    vector<int> res=Solution().findSubstring(s, words);

    if (res.size()> 0)

        for(int i=0; i<res.size(); i++)

        {

            if (i> 0)

                cout<<" ";

            cout<<res[i];

        }

    else

        cout<<-1;

 

    return 0;

}

2.输入说明

首先收入字符串s，

然后输入words中单词的数目n，

最后输入n个字符串，表示words中的单词

3.输出说明

按照索引由小到大顺序输出，或者输出-1.

4.范例

输入

barfoothefoobarman
2
foo bar

输出

0 9

5.代码

#include<iostream>
#include<map>
#include<string>
#include<unordered_map>
#include<algorithm>
#include<string.h>
#include<sstream>
#include <vector>

using namespace std;

vector<int> findSubstring(string s, vector<string> words)
{
    int word_len = words[0].size();//每个单词长度
    int s_len = s.size();
    int word_cnt = words.size();//总共有m个单词
    if (word_len == 0 || s_len == 0 || word_cnt == 0) return vector<int>{};

    unordered_map<string, int>word;//记录每个单词出现的次数
    for (auto it : words)
        word[it]++;

    vector<int> res;//结果
    
    unordered_map<string, int>tmp;
    
    int j;
    for (int i = 0; (i+ word_len*word_cnt ) <= s.size(); i++)//这里等号必须要加，否则会出错
    {
        for ( j = i; j < (i+word_len*word_cnt); j+=word_len)
        {
            string tmp_s = s.substr(j, word_len);//易错点1：substr(index,len) 第一个参数为下标，第二个为截取的长度
            //判断截取的子串是否存在于word中，且次数是否相等
            if (word[tmp_s] == 0)//不存在该单词，直接退出
                break;
            else
            {
                tmp[tmp_s]++;//哈希表更新
                if (word[tmp_s] < tmp[tmp_s])//重复次数过多，也要退出
                    break;
            }
        }
        
        if (j == (i + word_len * word_cnt))//遍历到末尾了
            res.push_back(i);    
        tmp.clear();//每次遍历结束都要清空临时哈希表
    }
    return res;
}
int main()

{

    string s, str;

    vector<string> words;

    int n;

    cin >> s;

    cin >> n;

    for (int i = 0; i < n; i++)

    {

        cin >> str;

        words.push_back(str);

    }

    vector<int> res = findSubstring(s, words);

    if (res.size() > 0)

        for (int i = 0; i < res.size(); i++)

        {

            if (i > 0)

                cout << " ";

            cout << res[i];

        }

    else

        cout << -1;



    return 0;

}