有道題目是這樣的:
輸入一個 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;
}