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

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 内完成所有位置的影响力代价计算。

http://www.xdnf.cn/news/3242.html

相关文章:

  • 如何在Cursor中使用MCP服务
  • Leetcode刷题记录24——最大子数组和
  • Java SE(6)——类和对象
  • 数据库 AI 助手测评:Chat2DB、SQLFlow 等工具如何提升开发效率?
  • 手搓传染病模型(SEIAR)
  • python3GUI--视频监控管理平台 By:PyQt5(详细讲解)
  • 多商户商城系统开发全策略:从技术架构到流量增长
  • python如何word转pdf
  • Node.js心得笔记
  • 前端八股 6
  • Redis ⑧-RESP | 渐进式遍历 | 数据库管理
  • 移动光猫 UNG853H 获取超级管理员账号密码
  • 2025东三省C题深圳杯C题数学建模挑战赛数模思路代码文章教学: 分布式能源接入配电网的风险分析
  • 支持selenium的chrome driver更新到136.0.7103.49
  • 2025年一加7pro刷twpr / magisk / kali nethunter教程+资源下载+避坑指南
  • ZYNQ 纯PL端逻辑资源程序固化流程
  • AM剪辑软件汉化版:简单易用,开启视频创作之旅
  • 5G技术如何提升智能家居体验:让家更聪明,生活更智能
  • Kubernetes(k8s)的API Server 组件原理与结合生产实战教程
  • 上位机知识篇---ARM 汇编语言与寄存器深度讨论
  • 【工具】Windows批量文件复制教程:用BAT脚本自动化文件管理
  • UDP数据包和TCP数据包的区别;网络编程套接字;不同协议的回显服务器
  • Android短信监控技术实现:合法合规的远程采集方案
  • 计网_PPP协议
  • CMake中的“包管理“模块FetchContent
  • Android Kotlin 项目完整集成 Bugly 异常监控指南
  • react学习笔记4——React UI组件库与redux
  • 《数据结构初阶》【顺序表/链表 精选15道OJ练习】
  • Python 与 MongoDB 深度融合:全流程数据库操作指南
  • 二、OrcaSlicer用户预设