有道题目是这样的:
输入一个 n 行 m 列网格,找一个格子,使得它所在的行和列中所有格子的数之和最大。如果答 案不唯一,输出任意解即可。比如,在下面的例子中,最优解是(1,3),即第一行和的三列的交 点(行从上到下编号为 1~n,列从左到右编号为 1~m),所有 7 个数之和为 35。
快要比赛的时候,有一个裁判想到了这样一个算法:
首先找一行 r(1<=r<=n) 使得该行所有数之和最大,然后找一列 c(1<=c<=m) 使得该列 所有数之和最大,最后直接输出(r,c)。如果有多个满足条件的 r,输出最小的 r。对 于 c 同样处理。
显然,这个算法是错的,但它竟然通过了大部分测试数据!你能找出那些让这个错误算法得到 正确结果的“弱”数据,以便裁判们改进这些数据吗?
Input
输入包含不超过 100 组数据。每组数据第一行为两个整数 n, m (1<=n<=500, 1<=m<=500),即行 数和列数。以下 n 行每行包含 m 个 1~100 的整数。输入的总大小不超过 2MB。
Output
对于每组数据,如果错误算法能得到正确结果,输出”Weak”,否则输出”Strong”。
Sample Input
4 4
5 5 5 5
1 1 5 1
1 1 5 1
1 1 5 1
5 4
2 5 1 1
1 1 9 1
1 1 1 1
1 1 1 1
1 1 1 1
Sample Output
Case 1: Weak
Case 2: Strong
把每行每列的和都算出来,再按照题意模拟两种方法即可
#include <iostream>
#include <cstring>
#include <string>
#include <vector>
#include <queue>
#include <cstdio>
#include <set>
#include <math.h>
#include <algorithm>
#include <queue>
#include <iomanip>
#define INF 0x3f3f3f3f
#define MAXN 10000005
#define Mod 1000000007
using namespace std;
int a[600][600],b[600][600],num[600][600];
int main()
{
int n,m;
int cnt=1;
while(~scanf("%d%d",&n,&m))
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
int ans1,maxnum1=-INF,maxnum2=-INF;
for(int i=1; i<=n; ++i)
for(int j=1; j<=m; ++j)
{
scanf("%d",&num[i][j]);
a[i][j]=a[i-1][j]+num[i][j];
b[i][j]=b[i][j-1]+num[i][j];
}
int ai;
for(int i=1; i<=m; ++i)
{
if(a[n][i]>maxnum1)
{
maxnum1=a[n][i];
ai=i;
}
}
int aj;
for(int i=1; i<=n; ++i)
{
if(b[i][m]>maxnum2)
{
maxnum2=b[i][m];
aj=i;
}
}
ans1=maxnum1+maxnum2-num[aj][ai];
int ans2=-INF;
for(int i=1; i<=n; ++i)
for(int j=1; j<=m; ++j)
ans2=max(a[n][j]+b[i][m]-num[i][j],ans2);
if(ans1==ans2)
printf("Case %d: Weak\n",cnt++);
else
printf("Case %d: Strong\n",cnt++);
}
return 0;
}