牛客小白月赛 22 G. 仓库选址

链接

https://ac.nowcoder.com/acm/contest/4462/G

题意

给出一个 m × n m \times n m×n 的矩阵,每个点都有一个访问次数,找出以某点为起点到其他点的距离 ∗ * 访问次数之和最小

思路

前缀和

若仓库在点 ( x , y ) (x,y) (x,y) 处,现在将仓库移动到点 ( x + 1 , y ) (x+1,y) (x+1,y) ,那么对于左上角为 ( 1 , 1 ) (1,1) (1,1),右下角为 ( x , m ) (x,m) (x,m) 的矩阵,所有点到达仓库的距离都 + 1 +1 +1,对于左上角为 ( x + 1 , 1 ) (x+1,1) (x+1,1),右下角为 ( n , m ) (n,m) (n,m) 的矩阵,所有点到达仓库的距离都 − 1 -1 1。若仓库在点 ( x , y ) (x,y) (x,y) 处,现在将仓库移动到点 ( x , y + 1 ) (x,y+1) (x,y+1) 同理计算即可

代码

#include<bits/stdc++.h>
using namespace std;
const int N=105;
int n,m;
int s[N][N];
int calc(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() {
	ios::sync_with_stdio(false);
    cin.tie(0);
	int T;cin>>T;
	while(T--) {
		cin>>m>>n;
		int now=0;
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++) {
				int x;
				cin>>x;
				now+=x*(i-1+j-1);
				s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+x;
			}
		}
		int res=0x3f3f3f3f;
		for(int i=1;i<=n;i++) {
			for(int j=1,t=now;j<=m;j++) {
				res=min(res,t);
				t+=calc(1,1,n,j)-calc(1,j+1,n,m);
			}
			now+=calc(1,1,i,m)-calc(i+1,1,n,m);
		}
		cout<<res<<endl;
	} 
 	return 0;
}
©️2020 CSDN 皮肤主题: 创作都市 设计师:CSDN官方博客 返回首页