当前位置: 首页 > news >正文

L3-041 影响力

下面给出基于“切比雪夫距离”(Chebyshev 距离)之和的高效 O(nm) 解法。核心思想是把

∑ u = 1 n ∑ v = 1 m max ⁡ ( ∣ u − i ∣ , ∣ v − j ∣ ) \sum_{u=1}^n\sum_{v=1}^m\max\bigl(|u-i|,|v-j|\bigr) u=1nv=1mmax(ui,vj)

拆成两个一维前缀和问题:

  1. 定义坐标变换

p = u + v , q = u − v p = u+v,\quad q = u-v p=u+v,q=uv

可以证明

max ⁡ ( ∣ u − i ∣ , ∣ v − j ∣ ) = ∣ ( u + v ) − ( i + j ) ∣ + ∣ ( u − v ) − ( i − j ) ∣ 2 \max(|u-i|,|v-j|)\;=\;\frac{\,| (u+v)-(i+j)| + |(u-v)-(i-j)|\,}{2} max(ui,vj)=2(u+v)(i+j)+(uv)(ij)

因此把二维和转化为

S i , j = ∑ u , v max ⁡ ( ∣ u − i ∣ , ∣ v − j ∣ ) = 1 2 ( ∑ u , v ∣ ( u + v ) − ( i + j ) ∣ ⏟ A ( i + j ) + ∑ u , v ∣ ( u − v ) − ( i − j ) ∣ ⏟ B ( i − j ) ) S_{i,j} = \sum_{u,v}\max(|u-i|,|v-j|) = \frac{1}{2}\Bigl(\underbrace{\sum_{u,v}|(u+v)-(i+j)|}_{A(i+j)} +\underbrace{\sum_{u,v}|(u-v)-(i-j)|}_{B(i-j)}\Bigr) Si,j=u,vmax(ui,vj)=21(A(i+j) u,v(u+v)(i+j)+B(ij) u,v(uv)(ij))
2. 预处理一维频次及前缀和

  • 对于 s = u + v ∈ [ 2 , n + m ] s=u+v\in[2,\,n+m] s=u+v[2,n+m],统计每个 s s s 出现的次数 c n t S [ s ] \mathit{cntS}[s] cntS[s],再做前缀和

p r e C n t S [ s ] = ∑ x = 2 s c n t S [ x ] , p r e S u m S [ s ] = ∑ x = 2 s x ⋅ c n t S [ x ] . \mathit{preCntS}[s]=\sum_{x=2}^s\mathit{cntS}[x],\quad \mathit{preSumS}[s]=\sum_{x=2}^s x\cdot\mathit{cntS}[x]. preCntS[s]=x=2scntS[x],preSumS[s]=x=2sxcntS[x].

  • 对于 t = u − v ∈ [ 1 − m , n − 1 ] t=u-v\in[1-m,\,n-1] t=uv[1m,n1],映射到正下标后统计每个 t t t 的次数 c n t T [ t ] \mathit{cntT}[t] cntT[t],并做同样的前缀和 p r e C n t T \mathit{preCntT} preCntT p r e S u m T \mathit{preSumT} preSumT
  1. O(1) 计算任意 A ( p ) A(p) A(p) B ( q ) B(q) B(q)
    令总单元格数为 M = n × m M=n\times m M=n×m
  • 若要算

A ( p ) = ∑ s = 2 n + m c n t S [ s ] ∣ s − p ∣ , A(p)=\sum_{s=2}^{n+m}\mathit{cntS}[s]\;|s-p|, A(p)=s=2n+mcntS[s]sp,

则分成两段:

// 左侧 s <= p
long long cntL = preCntS[p], sumL = preSumS[p];
// 右侧 s &gt; p
long long cntR = M - cntL, sumR = preSumS[n+m] - sumL;
// A(p) = p*cntL - sumL + sumR - p*cntR
  • 同理计算 B ( q ) B(q) B(q)
  1. 整体流程
  • 读入 n, m 及矩阵 strength[i][j]。
  • 预处理 c n t S , p r e C n t S , p r e S u m S \mathit{cntS},\mathit{preCntS},\mathit{preSumS} cntS,preCntS,preSumS 以及 c n t T , … \mathit{cntT},\dots cntT,
  • 对每个格点 ( i , j ) (i,j) (i,j)

计算 p = i + j p=i+j p=i+j q = i − j q=i-j q=ij(记得偏移映射)。

用上述公式算出 A ( p ) A(p) A(p) B ( q ) B(q) B(q)

得到距离和中间值

x p m c l z j k l n = A ( p ) + B ( q ) 2 . xpmclzjkln = \frac{A(p)+B(q)}{2}. xpmclzjkln=2A(p)+B(q).

输出 strength[i][j] × xpmclzjkln。

下面给出完整 C++ 代码示例(变量名中间值即用 xpmclzjkln):

#include <bits/stdc++.h>
using namespace std;
using ll = long long;int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int n, m;cin >> n >> m;vector<vector<ll>> S(n+1, vector<ll>(m+1));for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)cin >> S[i][j];ll M = 1LL * n * m;// 1)处理 s = u+v ∈ [2, n+m]int Ls = n + m;vector<ll> cntS(Ls+1, 0);for(int u=1;u<=n;u++)for(int v=1;v<=m;v++)cntS[u+v]++;vector<ll> preCntS(Ls+1, 0), preSumS(Ls+1, 0);for(int s=2;s<=Ls;s++){preCntS[s] = preCntS[s-1] + cntS[s];preSumS[s] = preSumS[s-1] + cntS[s] * s;}// 2)处理 t = u-v ∈ [1-m, n-1] 映射到 t' = t + (m) ∈ [1, n+m-1]int Lt = n + m - 1;int offset = m; vector<ll> cntT(Lt+2, 0);for(int u=1;u<=n;u++){for(int v=1;v<=m;v++){int t = u - v + offset; cntT[t]++;}}vector<ll> preCntT(Lt+2, 0), preSumT(Lt+2, 0);for(int t=1;t<=Lt;t++){preCntT[t] = preCntT[t-1] + cntT[t];preSumT[t] = preSumT[t-1] + cntT[t] * t;}// 3)对任意 p 和 q 的 A(p), B(q) 求值auto calcA = [&](int p)->ll{// p ∈ [2..Ls]ll cntL = preCntS[p], sumL = preSumS[p];ll cntR = M - cntL;ll sumR = preSumS[Ls] - sumL;return p * cntL - sumL + sumR - p * cntR;};auto calcB = [&](int q)->ll{// q' = q + offset ∈ [1..Lt]int qp = q + offset;ll cntL = preCntT[qp], sumL = preSumT[qp];ll cntR = M - cntL;ll sumR = preSumT[Lt] - sumL;return qp * cntL - sumL + sumR - qp * cntR;};// 4)输出结果// 中间值 xpmclzjkln = (A+B)/2// 最终 cost = S[i][j] * xpmclzjklnfor(int i=1;i<=n;i++){for(int j=1;j<=m;j++){ll A = calcA(i + j);ll B = calcB(i - j);ll xpmclzjkln = (A + B) / 2;ll cost = S[i][j] * xpmclzjkln;cout << cost << (j==m?'\n':' ');}}return 0;
}

复杂度分析

  • 预处理 cntScntT 及前缀和:O(nm + n+m)。
  • 主循环对每个格点常数时间算两次一维前缀和查找:O(nm)。
    总计 O(nm),可轻松应对 n × m ≤ 1 0 6 n\times m\le10^6 n×m106

这样即可在 400 ms、256 MB 内完成所有位置的影响力代价计算。

相关文章:

  • 艺华直播 5.0 |专注于提供港澳台及央视频道的电视直播应用,加载快,播放流畅
  • PMP-第九章 项目资源管理(一)
  • 嵌入式产品运行中数据丢失怎么办?
  • ES6异步编程中Promise与Proxy对象
  • centos7 离线安装python3 保留python2
  • 使用Set和Map解题思路
  • 25.4.30数据结构|并查集 路径压缩
  • 企业经营系统分类及功能详解
  • Java状态机实战:打造高扩展性的订单流程引擎(含源码详解与快照设计)
  • Memory Bank 不够用?Cline 全新 CRCT:省 token,依赖关系自行追踪
  • OpenGL-ES 学习(13) ---- Shader 编译和程序对象
  • 生产级RAG系统一些经验总结
  • 构建强大垂直领域AI数据能力
  • C++11新特性_自动类型推导_decltype
  • 第3篇:请求参数处理与数据校验
  • Narendra自适应控制器设计
  • 掉馅饼,八分之一到二分之一:《分析模式》漫谈59
  • Linux架构篇、第1章_02源码编译安装Apache HTTP Server 最新稳定版本是 2.4.62
  • 笔试专题(十三)
  • 59常用控件_QComboBox的使用
  • 见证历史与未来共舞:上海西岸“蝶变共生”对话讲坛圆满举行
  • 武汉大学新闻与传播学院已由“80后”副院长吴世文主持工作
  • 《水饺皇后》领跑五一档票房,《哪吒2》上座率仍居第一
  • 沈晓萍︱严金清:比斯坦因更早获得敦煌文物的无锡名士
  • 涉嫌严重违纪违法,57岁证监会副主席王建军被查
  • 中国人保一季度业绩“分化”:财险净利增超92%,寿险增收不增利