Code bài AMESSAGE bị sai

[Hien_Nguyen (Hiển Nguyễn)]

Haizz, luyện tập code trên NTU chán thật đấy, code mẫu chẳng có mà test 2 trở đi số đã to rồi, không xem được hết test.

Đề bài gốc :

http://ntucoder.net/Problem/Details/5520
Tại năm 2060, Khoa là một nhà khoa học lừng danh. Ông ta rất thích nghiên cứu về ngôn ngữ. Trong một lần phóng tên lửa lên Mặt Trăng, máy tính của ông tình cờ nhận được một tệp tin và M tín hiệu rất kì lạ. Tệp tin chứa N dãy nhị phân và M tín hiệu nhận được cũng là những dãy nhị phân. Khoa cho rằng đây chính là tín hiệu của những người ngoài hành tinh đang gửi cho nhau và bức thông điệp kia có thể là một cuốn từ điển để giải mã các tín hiệu ấy. Với bản tính say mê khoa học của mình, Khoa muốn đếm xem với mỗi dãy nhị phân có trong tệp tin nhận được, có bao nhiêu tín hiệu có phần đầu khớp với phần đầu của dãy nhị phân đó, hay nói cách khác: Nếu độ dài tín hiệu nhỏ hơn hoặc bằng độ dài dãy nhị phân đó thì dãy nhị phân đó phải bắt đầu bằng tín hiệu, ngược lại thì tín hiệu phải bắt đầu bằng dãy nhị phân đó. Tuy nhiên dữ liệu được gửi về có thể rất lớn nên Khoa cần một chương trình để thực hiện công việc này. Bạn hãy giúp Khoa nhé!

Input

Dòng đầu chứa 2 số nguyên M, N (1 <= M, N <= 50000)

M dòng tiếp theo, mỗi dòng ghi một số L (1 <= L <= 10000) là độ dài của mỗi dãy nhị phân trong các tín hiệu, tiếp theo là L kí tự 0 hoặc 1 phân cách với nhau bởi một khoảng trắng là dãy nhị phân đó.

N dòng tiếp theo, mỗi dòng ghi một số L (1 <= L <= 10000) là độ dài của mỗi dãy nhị phân trong tệp tin, tiếp theo là L kí tự 0 hoặc 1 phân cách với nhau bởi một khoảng trắng là dãy nhị phân đó.

Lưu ý: Tổng độ dài các dãy nhị phân trong input không vượt quá 500000

Output

Gồm N dòng, mỗi dòng in ra số tín hiệu khớp với phần đầu của dãy nhị phân đó.

Ví dụ

• input

4 5
3 0 1 0
1 1
3 1 0 0
3 1 1 0
1 0
1 1
2 0 1
5 0 1 0 0 1
2 1 1

output

1
3
1
1
2

Giải thích ví dụ

Có 4 tín hiệu nhận được là 010, 1, 100 và 110

Có 5 dãy nhị phân trong tập tin là 0, 1, 01, 01001, và 11

Dãy 0 khớp với phần đầu của tín hiệu 010 nên kết quả là 1

Dãy 1 khớp với phần đầu của tín hiệu 1, 100 và 110 nên kết quả là 3

Dãy 01 khớp với phần đầu của tín hiệu 010 nên kết quả là 1

Dãy 01001 khớp với tín hiệu 010 nên kết quả là 1

Dãy 11 khớp với tín hiệu 1 và 110 nên kết quả là 2

Bài này là cho m dãy nhị phân loại 1 và n dãy nhị phân loại 2 (mỗi dãy sẽ có độ dài chuỗi, và từng c.số cách nhau 1 dấu cách). Với mỗi dãy A là loại 2, cho biết có bao nhiêu xâu nhận bao nhiêu xâu A là tiền tố, hoặc là xâu tiền tố của A?

Em không biết mình sai chỗ nào, mong mọi người giúp với ạ.
Em sử dụng Trie ạ

#include <bits/stdc++.h>
#define ll long long
#define maxn 500005

using namespace std;

struct node{
    int deg;
    int child[2];
    int numEnd;
};

int n, m, cnt;
node T[maxn];

int main(){
    faster
    cin >> m >> n;
    while (m--){
        int L, i, x = 0;
        cin >> L;
        while (L--){
            cin >> i;
            if (T[x].child[i] == 0) T[x].child[i] = ++cnt;
            x = T[x].child[i];
            T[x].deg++;
        }
        T[x].numEnd++;
    }

    while (n--){
        int L, i, x = 0, res = 0;
        cin >> L;
        while (L--){
            cin >> i;
            if (T[x].child[i] == 0) break;
            res += T[x].numEnd;
            x = T[x].child[i];
        }
        if (L > 0) while (L--) cin >> i;
        cout << res + T[x].deg << '\n';
    }
}