Codeforces Round 790 (Div. 4) D
做题链接:https://codeforces.com/contest/1676/problem/D
帖木儿的爷爷送给他一个棋盘,让他练习棋艺。这个棋盘是一个a行n列的网格,每个单元格都写有一个非负整数。
帖木儿的挑战是在棋盘上放置一只主教,使主教攻击的所有单元格之和为最大。主教可以向对角线的所有方向攻击,而且攻击距离没有限制。请注意,主教所在的单元格也被视为受到攻击。帮助他找出他能得到的最大和。
输入
输入的第一行包含一个整数 t(1≤t≤1000)--测试用例数。测试用例说明如下。
每个测试用例的第一行包含整数 n和 m(1≤n≤200,1≤m≤200)。
下面的n行各包含m个整数,i-行的j-个元素aij是写在j-行的i-个单元格中的(0≤aij≤10^6)(0≤10^6)个数字。
保证所有测试用例的 n⋅m之和不超过 4⋅10^4。
输出
对每个测试用例输出一个整数,即所有可能的主教位置的最大和。
思想:暴力,但是要注意很多细节,每次把当前点要走的下一步作为起点开始斜线遍历,这样可以有效避免越界问题,并且最后只用加上当前数。
代码:
// Problem: D. X-Sum
// Contest: Codeforces - Codeforces Round 790 (Div. 4)
// URL: https://codeforces.com/contest/1676/problem/D
// Memory Limit: 256 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 210;
int main(){
int T;
cin>>T;
while(T--){
int n,m;
cin>>n>>m;
int a[N][N];
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>a[i][j];
}
}
int maxv=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
int ans=0;
int x=i+1,y=j+1;
while(x<=n&&y<=m){
ans+=a[x][y];
x++;
y++;
}
x=i-1,y=j-1;
while(x>=1&&y>=1){
ans+=a[x][y];
x--;
y--;
}
x=i+1,y=j-1;
while(x<=n&&y>=1){
ans+=a[x][y];
x++;
y--;
}
x=i-1,y=j+1;
while(x>=1&&y<=m){
ans+=a[x][y];
x--;
y++;
}
ans+=a[i][j];
maxv=max(maxv,ans);
}
}
cout<<maxv<<"\n";
}
return 0;
}