#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"); } |
English