Kmp算法的变题

字符串包含

字符串包含
时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 32M，其他语言64M 热度指数：9356
本题知识点： Java工程师 C++工程师 快手 字符串 模拟 游戏研发工程师 快手 2019
算法知识视频讲解
校招时部分企业笔试将禁止编程题跳出页面，为提前适应，练习时请使用在线自测，而非本地IDE。
题目描述
我们定义字符串包含关系：字符串A=abc，字符串B=ab，字符串C=ac，则说A包含B，A和C没有包含关系。
输入描述:

两个字符串，判断这个两个字符串是否具有包含关系，测试数据有多组，请用循环读入。

输出描述:

如果包含输出1，否则输出0.

示例1
输入
复制

abc ab

输出
复制

1

import java.util.*;
public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()) {
             String A = sc.next();
            String B = sc.next();
            boolean c = c = A.contains(B)||B.contains(A);

            if(c) {
                System.out.println(1);
            }else{
                System.out.println(0);
            }
        }
    }
}

链接：https://www.nowcoder.com/questionTerminal/661e24c11de64e78804fdce653dafb0e?f=discussion
来源：牛客网

#include <iostream>
#include <vector>
using namespace std;
void getNext(string str, vector<int>& next);
int kmp(string str1, string str2);
int main(){
 
    string str1, str2;
    while(cin >> str1 >> str2){
        if(kmp(str2, str1) != -1 || kmp(str1, str2) != -1 ){
            cout << "1" << endl;
        }else{
            cout << "0" << endl;
        }
    }
 
    return 0;
}
void getNext(string str, vector<int>& next){
    int n = str.size();
    next[0] = -1;
    next[1] = 0;
    int cn = 0;
    int i = 2;
    while(i < next.size()){
        if(str[i - 1] == str[cn]){
            next[i++] = ++cn;
        }else if(cn > 0){
            cn = next[cn];
        }else{
            next[i++] = 0;
        }
    }
}
int kmp(string str1, string str2){
 
    int n = str1.size();
    vector<int> next(n, -1);
    getNext(str1, next);
    int i1 = 0, i2 = 0;
    while(i1 < str1.size() && i2 <= str2.size()){
        if(str1[i1] == str2[i2]){
            ++i1, ++i2;
        }else if(i1 == 0){
            ++i2;
        }else{
            i1 = next[i1];
        }
    }
    return i1 == str1.size() ? i2 - i1 : -1;
}