題面

小 y 終於在小學學會了一些字、詞、句子， 會寫一點作文了。某一天，小 y 買 了一張方格稿紙來寫作文， 稿紙是 n 行 m 列的，形狀如下所示（圖中 n=m=5）：

某天小 y 的鄰居小 x 來小 y 家玩， 無聊地用黑墨水筆把小 y 新買的方格稿紙 塗黑了很多格子。每個格子不是完全黑色就是完全白色，如下圖所示。

小 y 不能責怪小 x。作文寫不成了，他也覺得很無聊，就開始數裡面有多少 “魔 幻方陣”。如果稿紙中一個 k×k 的正方形區域滿足以下兩個條件，那麼它就是魔幻 方陣：

（1） 黑白格子的數量差不能超過 1； （2） k 不能小於 2。 上圖染色後的方格稿紙共有 9 個魔幻方陣（6 個 2×2 的魔幻方陣， 3 個 3×3 的魔幻方陣）。

現在，請你幫助小 y 程式設計計算被染色的稿紙裡面有多少個魔幻方陣。

對於 50%的資料滿足： 1≤n≤10， 1≤m≤10； 對於 75%的資料滿足： 1≤n≤180， 1≤m≤180； 對於 100%的資料滿足： 1≤n≤300， 1≤m≤300。

分析

n³,二維字首和隨便亂搞。

程式碼

#include<bits/stdc++.h>
using namespace std;
#define N 303
int w,b,n,m,ans;
int a[N][N],white[N][N],black[N][N];
inline void read(int &x)
{
    x
 
=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
}
int main()
{
    read(n),read(m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            read(a[i][j]);
            white[i][j]
 
=white[i-1][j]+white[i][j-1]-white[i-1][j-1]+(a[i][j]?0:1);
            black[i][j]=black[i-1][j]+black[i][j-1]-black[i-1][j-1]+(a[i][j]?1:0);
        }        
    for(int k=2;k<=min(n,m);k++)
        for(int i=k;i<=n;i++)
            for(int j=k;j<=m;j++)
            {
                w=white[i][j]-white[i-k][j]-white[i][j-k]+white[i-k][j-k];
                b=black[i][j]-black[i-k][j]-black[i][j-k]+black[i-k][j-k];
                if(abs(w-b)<=1)ans++;
            }    
    printf("%d\n",ans);
    return 0;
}