csu1779——错误的算法（模拟）

有道题目是这样的:

输入一个 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;
}