仓库选址

传送门

题目描述
牛能在某小城有了固定的需求，为了节省送货的费用，他决定在小城里建一个仓库，但是他不知道选在哪里，可以使得花费最小。
给出一个m×n的矩阵，代表下一年小城里各个位置对货物的需求次数。我们定义花费为货车载货运输的距离，货车只能沿着水平或竖直方向行驶。

输入描述:
首先在一行中输入T ,T≤10，代表测试数据的组数。
每组输入在第一行给出两个正整数n, m,1≤n,m≤100，分别代表矩阵的宽和高。
接下来m行，每行n个不超过1000的数字，代表矩阵里的元素。
输出描述:
每组输入在一行中输出答案。

思路：
由于时限给了4s，所以暴力能过。
正确解法：我们假设仓库坐标（x,y）,当仓库移动到（x，y+1）时，那么左上角（1，1）到右下角（n,y）的子矩阵移动距离+1；左上角（x，y+1）到右下角（n，m）的子矩阵移动距离-1。同样考虑（x，y）移动到（x+1，y）的情况。
我们首先暴力处理出仓库在（1，1）花费，再利用二维前缀和处理运输次数，接着就是枚举转移。

#define first f
#define second s
#define ll long long
#define mp make_pair
#define pb push_back
#define pf push_front
#define lb lower_bound
#define ub upper_bound
#include <bits/stdc++.h>
#define pii pair<int,int>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=1e5+5;
const int MOD=1e9+7;
const double PI=acos(-1);
const double e=2.718281828459;

int n,m;
int ma[105][105],ans[105][105];
int cal(int x2,int y2,int x1,int y1)
{
    return ma[x2][y2]-ma[x2][y1-1]-ma[x1-1][y2]+ma[x1-1][y1-1];
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&m,&n);
        ans[1][1]=0;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                scanf("%d",&ma[i][j]);
                ans[i][j]=0;
                ans[1][1]+=ma[i][j]*(i+j-2);
            }
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                ma[i][j]=ma[i-1][j]+ma[i][j-1]-ma[i-1][j-1]+ma[i][j];
            }
        }
        int cnt=ans[1][1];
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                if(i==1&&j==1){continue;}
                if(j!=1){
                    ans[i][j]=ans[i][j-1]+cal(n,j-1,1,1)-cal(n,m,1,j);
                }
                else{
                    ans[i][j]=ans[i-1][j]+cal(i-1,m,1,1)-cal(n,m,i,j);
                }
                cnt=min(cnt,ans[i][j]);
            }
        }
        printf("%d\n",cnt);
    }
    return 0;
}