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;
}