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

最新推荐文章于 2024-05-04 20:17:58 发布

c4Lnn

最新推荐文章于 2024-05-04 20:17:58 发布

阅读量192

点赞数

分类专栏：基础 —— 前缀和、差分

本文链接：https://blog.csdn.net/qq_43651414/article/details/107678182

版权

基础 —— 前缀和、差分专栏收录该内容

10 篇文章 0 订阅

订阅专栏

链接

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

题意

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

思路

前缀和

若仓库在点 $(x, y)$ 处，现在将仓库移动到点 $(x + 1, y)$ ，那么对于左上角为 $(1, 1)$ ，右下角为 $(x, m)$ 的矩阵，所有点到达仓库的距离都 $+ 1$ ，对于左上角为 $(x + 1, 1)$ ，右下角为 $(n, m)$ 的矩阵，所有点到达仓库的距离都 $- 1$ 。若仓库在点 $(x, y)$ 处，现在将仓库移动到点 $(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;
}