Cho mảng $A$ gồm $m$ phần tử (các phần tử được đánh số từ $1$ đến $m$) và mảng $B$ gồm $n$ phần tử (các phần tử được đánh số từ $1$ đến $n$), mỗi phần tử của hai mảng chỉ nhận một trong ba giá trị $-1,0,1$. Tiến hành tạo bảng $C$ kích thước $m\times n$, trong đó phần tử $(i,j)$ nhận giá trị $C[i][j]=A[i]\times B[j]$ với $1\le i\le m$ và $1\le j\le n$.

Một bảng con vuông kích thước $s$ của bảng $C$ có phần tử trái trên là $(x,y)$ và phần tử phải dưới là $(x+s-1,y+s-1)$ với $1\le x\le m-s+1$ và $1\le y\le n-s+1$ được gọi là bảng con vuông “cân bằng” nếu:

• Các phần tử thuộc đường chéo chính đều nhận giá trị bằng $1$. Cụ thể, các phần tử $(x,y),(x+1,y+1),\dots,(x+s-1,y+s-1)$ có giá trị bằng $1$.
• Tổng tất các các phần tử trong bảng con vuông bằng $0$.

Yêu cầu: Cho hai mảng $A$ và $B$, đếm số bảng con vuông “cân bằng” trong bảng $C$.

3 4
1 -1 1
1 0 -1 1

1