仓库选址(二维前缀和,距离和问题)

仓库选址

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

输入描述:

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

输出描述:

每组输入在一行中输出答案。

输入

3
2 2
1 1
1 0
4 4
0 8 2 0
1 4 5 0
0 1 0 1
3 9 2 0
6 7
0 0 0 0 0 0
0 1 0 3 0 1
2 9 1 2 1 2
8 7 1 3 4 3
1 0 2 2 7 7
0 1 0 0 1 0
0 0 0 0 0 0

输出

2
55
162

备注:

送货时只能单次运输，若该位置需要3次，货车必须跑3次。
即使该位置需要被送货，我们仍然可以选择该位置作为仓库。

算法分析

这道题的数据比较弱,可以暴力解决,正解的话,会用到二维前缀和,二维前缀和的做法比如我们当前在(x,y)选为仓库,当(x,y+1)选为仓库的时候,那么左上角为(1,1)到右下角(n,y)的子矩阵到该点的距离都增加了 1,而左上角为(x,y+1)到右下角(n,m)的子矩阵就减少了 1.减少或增加的总距离就是子矩阵的前缀和.所以我们算出(1,1)到(n,m)的代价和,之后移动点,算出最小距离

代码实现

暴力做法

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<math.h>
using namespace std;
const int maxn=1e2+5;
typedef long long ll;
int t,n,m;
int map[maxn][maxn];
int dis,res;
int slove(int x,int y)
{
    dis=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            dis+=map[i][j]*(abs(x-i)+abs(y-j));
        }
    }
    return dis;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        res=0x3f3f3f3f;
        scanf("%d%d",&m,&n);
        memset(map,0,sizeof(map));
        for(int i=1;i<=n;i++)
          for(int j=1;j<=m;j++)
              scanf("%d",&map[i][j]);
        for(int i=1;i<=n;i++)
           for(int j=1;j<=m;j++)
           {
              int a=slove(i,j);
              res=min(res,a);
           }
        cout<<res<<endl;
    }
    return 0;
}

二维前缀和做法

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<math.h>
using namespace std;
const int maxn=1e2+5;
typedef long long ll;
int t,n,m;
int map[maxn][maxn];
int s[maxn][maxn];
int dis,res;
int col(int a,int b,int c,int d)
{
    return s[c][d]-s[a-1][d]-s[c][b-1]+s[a-1][b-1];
}
int main()
{
    cin>>t;
    while(t--)
    {
        int tot=0;
        cin>>m>>n;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            {
                cin>>map[i][j];
                tot+=map[i][j]*(i-1+j-1);
                s[i][j]=s[i][j-1]+s[i-1][j]+map[i][j]-s[i-1][j-1];
            }
        int res=0x3f3f3f3f;
        int tt;
        for(int i=1;i<=n;i++)
        {
            tt=tot;
            res=min(res,tt);
            for(int j=1;j<=m;j++)
            {
                res=min(tt,res);
                tt+=col(1,1,n,j)-col(1,j+1,n,m);
            }
            tot+=col(1,1,i,m)-col(i+1,1,n,m);
        }
        cout<<res<<endl;
    }
}