codeforces - 738B - Spotlights （瞎搞）

codeforces - 738B - Spotlights （瞎搞）

#include<bits/stdc++.h>
using namespace std;
int maze[1010][1010];
int can[1010][1010]={0};
bool col[1010]={0},row[1010]={0};
bool col2[1010]={0},row2[1010]={0};
int n,m;
int main()
{
cin>>n>>m;
for(int i=0; i<n; i++)
{
for(int j=0; j<m; j++)
{
scanf("%d",&maze[i][j]);
if(maze[i][j]==0)
{
if(col[i])can[i][j]++;
if(row[j])can[i][j]++;
}
else
{
col[i]=1;
row[j]=1;
}
}
}

for(int i=n-1; i>=0; i--)
{
for(int j=m-1; j>=0; j--)
{
if(maze[i][j]==0)
{
if(col2[i])can[i][j]++;
if(row2[j])can[i][j]++;
}
else
{
col2[i]=1;
row2[j]=1;
}
}
}
__int64 ans=0;
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
if(maze[i][j]==0)
{
ans+=can[i][j];
}
}
}
cout<<ans<<endl;
return 0;
}