왜 안풀림

하..

https://www.acmicpc.net/problem/1786

1786번: 찾기

글 읽기 - 40% 쯤에서 틀립니다 문제가 뭔지 도저히 모르겠어요...

 

 

 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <set>
using namespace std;

void kmp(string t, string p, vector<int> &box) {
    int ans = 0;
    int i = 0, j = 0;
    set<int> pos;
    
    while (i < t.length()) {
        if (t[i] == p[j]) {
            i++;
            j++;
        } else {
            if (j == 0) {
                i++;
            } else {
                j = box[j-1];
            }
        }
        if (j == p.length()) {
            pos.insert(i - p.length() + 1);
            j = box[j-1];
            i = i - p.length() + 2;
            ans++;
        }
    }
    cout << ans << '\n';
    for (auto i : pos) {
        cout << i << " ";
    }
}

void get_index(string p, vector<int> &box) {
    int i = 0, j = 1;
    while (j < p.length()) {
        if (p[i] == p[j]) {
            box[j] = i+1;
            i++;
            j++;
        } else {
            if (i == 0) {
                j++;
            } else {
                i = box[i-1];
            }
        }
    }
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    
    string T, P;
    getline(cin, T);
    getline(cin, P);
    // if (T.at(T.size() - 1) == '\n') T.erase(T.end());
    vector<int> v(P.length(), 0);
    
    get_index(P, v);
    kmp(T, P, v);
    
    return 0;
}