#include <iostream> #include <stdio.h> #include <vector> #include <algorithm> using namespace std; int main() { int n, m, l, a, t = 0; scanf("%d%d", &n, &m); vector<int> v[n]; int wzorzec[m], P[m]; for (int i = 0; i < n; i++) { scanf("%d", &l); for (int j = 0; j < l; j++) { scanf("%d", &a); v[i].push_back(a); } } for (int i = 0; i < m; i++) scanf("%d", &wzorzec[i]); //obliczenie tablicy P z algorytmi KMP P[0] = 0; P[1] = 0; for (int j = 2; j <= m; j++) { while ((t>0) && (wzorzec[t] != wzorzec[j - 1])) t = P[t]; if (wzorzec[t] == wzorzec[j - 1]) t++; P[j] = t; } vector<int> tekst, pop; tekst.push_back(1); int pokolenie = 1; bool czyResult = false; while (tekst.size() < 50000000 && pokolenie < 1000000) { if (pokolenie > 1) { //generacja nowego pokolenia pop = tekst; tekst = vector<int>(); for (int ii = 0; ii < pop.size(); ii++) for (int jj = 0; jj < v[pop[ii] - 1].size(); jj++) tekst.push_back(v[pop[ii] - 1][jj]); } int i = 1, j = 0; if (i <= (int)tekst.size() - m + 1) while (i <= (int)tekst.size() - m + 1) { j = P[j]; while ((j<m) && (wzorzec[j] == tekst[i + j - 1])) j++; if (j == m) { czyResult = true; break; } if (1 > j - P[j]) i++; else i = i + j - P[j]; } if (czyResult) break; pokolenie++; } if (czyResult) printf("%d\n", pokolenie); else printf("-1\n"); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | #include <iostream> #include <stdio.h> #include <vector> #include <algorithm> using namespace std; int main() { int n, m, l, a, t = 0; scanf("%d%d", &n, &m); vector<int> v[n]; int wzorzec[m], P[m]; for (int i = 0; i < n; i++) { scanf("%d", &l); for (int j = 0; j < l; j++) { scanf("%d", &a); v[i].push_back(a); } } for (int i = 0; i < m; i++) scanf("%d", &wzorzec[i]); //obliczenie tablicy P z algorytmi KMP P[0] = 0; P[1] = 0; for (int j = 2; j <= m; j++) { while ((t>0) && (wzorzec[t] != wzorzec[j - 1])) t = P[t]; if (wzorzec[t] == wzorzec[j - 1]) t++; P[j] = t; } vector<int> tekst, pop; tekst.push_back(1); int pokolenie = 1; bool czyResult = false; while (tekst.size() < 50000000 && pokolenie < 1000000) { if (pokolenie > 1) { //generacja nowego pokolenia pop = tekst; tekst = vector<int>(); for (int ii = 0; ii < pop.size(); ii++) for (int jj = 0; jj < v[pop[ii] - 1].size(); jj++) tekst.push_back(v[pop[ii] - 1][jj]); } int i = 1, j = 0; if (i <= (int)tekst.size() - m + 1) while (i <= (int)tekst.size() - m + 1) { j = P[j]; while ((j<m) && (wzorzec[j] == tekst[i + j - 1])) j++; if (j == m) { czyResult = true; break; } if (1 > j - P[j]) i++; else i = i + j - P[j]; } if (czyResult) break; pokolenie++; } if (czyResult) printf("%d\n", pokolenie); else printf("-1\n"); } |