計蒜客模擬賽D1T1 蒜頭君打地鼠：矩陣旋轉+二維字首和

Leohh發表於2017-07-30
矩陣

題目連結：https://nanti.jisuanke.com/t/16445

題意：

　　給你一個n*n大小的01矩陣，和一個k*k大小的錘子，錘子只能斜著砸，問只砸一次最多能砸到多少個1。

　　　　　　　　　　　　　

 

題解：

　　將原矩陣順時針旋轉45°，二維字首和預處理，然後列舉每一個可能砸到的正方形之和並取最大。

　　注：列舉的正方形的四個頂點必須是從原矩陣旋轉過來的點，否則會出現砸到下面的這種情況：

　　　　　　　　　　　　

　　　　（*代表不是原矩陣旋轉過來的點，陰影代表砸到的部分）

　　　　程式碼中用vis陣列判斷是否為原矩陣中旋轉過來的點。

 

AC Code：

#include <iostream>
#include <stdio.h>
#include <string.h>
#define MAX_N 4005

using namespace std;

int n,k,t;
int ans;
int a[MAX_N][MAX_N];
int sum[MAX_N][MAX_N];
bool vis[MAX_N][MAX_N];

void read()
{
    memset(a,0,sizeof(a));
    memset(vis,false,sizeof(vis));
    scanf("%d%d",&n,&k);
    t=n;
    int temp;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            scanf("%d",&temp);
            a[i+j][n-1-i+j]=temp;
            vis[i+j][n-1-i+j]=true;
        }
    }
    n=n*2-1;
    k=k*2-1;
}

bool is_legal(int x,int y)
{
    return x+y>=t-1 && n-1-x+y>=t-1 && n-1-y+x>=t-1 && 2*n-2-x-y>=t-1 && vis[x][y];
}

void cal_sum()
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            if(i-1>=0) sum[i][j]+=sum[i-1][j];
            if(j-1>=0) sum[i][j]+=sum[i][j-1];
            if(i-1>=0 && j-1>=0) sum[i][j]-=sum[i-1][j-1];
            sum[i][j]+=a[i][j];
        }
    }
}

int cal_max()
{
    int maxn=0;
    for(int i=0;i<n && i+k-1<n;i++)
    {
        for(int j=0;j<n && j+k-1<n;j++)
        {
            if(!is_legal(i,j) || !is_legal(i,j+k-1) || !is_legal(i+k-1,j) || !is_legal(i+k-1,j+k-1)) continue;
            int now=sum[i+k-1][j+k-1];
            if(i-1>=0) now-=sum[i-1][j+k-1];
            if(j-1>=0) now-=sum[i+k-1][j-1];
            if(i-1>=0 && j-1>=0) now+=sum[i-1][j-1];
            maxn=max(maxn,now);
        }
    }
    return maxn;
}

void solve()
{
    cal_sum();
    ans=cal_max();
}

void print()
{
    printf("%d\n",ans);
}

int main()
{
    read();
    solve();
    print();
}

 

相關文章