题目描述
牛能在某小城有了固定的需求,为了节省送货的费用,他决定在小城里建一个仓库,但是他不知道选在哪里,可以使得花费最小。
给出一个m×n的矩阵,代表下一年小城里各个位置对货物的需求次数。我们定义花费为货车载货运输的距离,货车只能沿着水平或竖直方向行驶。
题目链接:https://ac.nowcoder.com/acm/problem/202492
思路:
首先计算出拿[1,1]为仓库需要跑的总距离,只有每次移动一格,来计算改变的总距离,在所有点中找到总距离最小的那个点。
在输入的循环中,now表示以[1,1]为起点的总距离,pre表示前缀矩阵。
每当检测的点右移一次,所有[1,1],[n,j]内的点,距离都加一,同时[1,j+1],[n,m]内的点,距离都减一。
每当检测的点下移一次,所有[1,1],[i,m]内的点,距离都加一,同时[i+1,1],[n,m]内的点,距离都减一。
在这些点中找到总距离最小的点。
思路参考:
https://ac.nowcoder.com/acm/problem/blogs/202492
代码:
#include<iostream>
using namespace std;
const int N=1e3;
int pre[N][N];
int calc(int a,int b,int c,int d)
{
return pre[c][d]-pre[c][b-1]-pre[a-1][d]+pre[a-1][b-1];
}
int main()
{
int T;
cin>>T;
while(T--)
{
int n,m;
cin>>m>>n;
int now=0,nxt=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
int t;
cin>>t;
now+=(i-1+j-1)*t;
pre[i][j]=pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]+t;
}
}
int ans=0x3f3f3f3f;
for(int i=1;i<=n;i++)
{
nxt=now;
for(int j=1;j<=m;j++)
{
ans=min(ans,nxt);
nxt+=calc(1,1,n,j)-calc(1,j+1,n,m);
}
now+=calc(1,1,i,m)-calc(i+1,1,n,m);
}
cout<<ans<<endl;
}
}