AT_agc066_a [AGC066A] Adjacent Difference题解

Link

对于数组中任意两个相邻的元素 x x x 和 y y y,如果满足 x ≡ d ( m o d 2 d ) x \equiv d \pmod{2 d} x≡d(mod2d) 且 y ≡ 0 ( m o d 2 d ) y \equiv 0 \pmod{2 d} y≡0(mod2d) 显然是合法的。于是我们可以将所有满足 ( i + j ) ≡ 1 ( m o d 2 ) (i+j) \equiv 1 \pmod{2} (i+j)≡1(mod2) 的 a i , j a_{i,j} ai,j 改到最近的满足 a i , j ≡ d ( m o d 2 d ) a_{i,j} \equiv d \pmod{2 d} ai,j≡d(mod2d) 的数,将所有满足 ( i + j ) ≡ 0 ( m o d 2 ) (i+j) \equiv 0 \pmod{2} (i+j)≡0(mod2) 的 a i , j a_{i,j} ai,j 改到最近的满足 a i , j ≡ 0 ( m o d 2 d ) a_{i,j} \equiv 0 \pmod{2 d} ai,j≡0(mod2d) 的数。或者,我们还可以将所有满足 ( i + j ) ≡ 0 ( m o d 2 ) (i+j) \equiv 0 \pmod{2} (i+j)≡0(mod2) 的 a i , j a_{i,j} ai,j 改到最近的满足 a i , j ≡ d ( m o d 2 d ) a_{i,j} \equiv d \pmod{2 d} ai,j≡d(mod2d) 的数,将所有满足 ( i + j ) ≡ 1 ( m o d 2 ) (i+j) \equiv1 \pmod{2} (i+j)≡1(mod2) 的 a i , j a_{i,j} ai,j 改到最近的满足 a i , j ≡ 0 ( m o d 2 d ) a_{i,j} \equiv 0 \pmod{2 d} ai,j≡0(mod2d) 的数。

证明以上两种方案的总代价是 n d 2 nd^2 nd2

若 0 ≤ ( a i , j   m o d   2 d ) ≤ d 0 \le (a_{i,j}\bmod 2d) \le d 0≤(ai,jmod2d)≤d,则改到最近的满足 a i , j ≡ 0 ( m o d 2 d ) a_{i,j} \equiv 0 \pmod{2 d} ai,j≡0(mod2d) 的数需要 ( a i , j   m o d   2 d ) (a_{i,j}\bmod 2d) (ai,jmod2d) 的代价,改到最近的满足 a i , j ≡ d ( m o d 2 d ) a_{i,j} \equiv d \pmod{2 d} ai,j≡d(mod2d) 的数需要 d − ( a i , j   m o d   2 d ) d-(a_{i,j}\bmod 2d) d−(ai,jmod2d) 的代价,两种方案的代价和为 d d d。

若 d < ( a i , j   m o d   2 d ) ≤ 2 d d < (a_{i,j}\bmod 2d) \le 2d d<(ai,jmod2d)≤2d,则改到最近的满足 a i , j ≡ 0 ( m o d 2 d ) a_{i,j} \equiv 0 \pmod{2 d} ai,j≡0(mod2d) 的数需要 2 d − ( a i , j   m o d   2 d ) 2d-(a_{i,j}\bmod 2d) 2d−(ai,jmod2d) 的代价,改到最近的满足 a i , j ≡ d ( m o d 2 d ) a_{i,j} \equiv d \pmod{2 d} ai,j≡d(mod2d) 的数需要 ( a i , j   m o d   2 d ) − d (a_{i,j}\bmod 2d)-d (ai,jmod2d)−d 的代价,两种方案的代价和为 d d d。

故对于数组中任意一个元素两种方案的代价和为 d d d,所以两种方案的总代价是 n d 2 nd^2 nd2。

故至少有一种方案的代价不超过 n d 2 2 \frac{nd^2}{2} 2nd2。

Code

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
int n,d,a[505][505];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n>>d;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++) cin>>a[i][j];
	int s1=0,s2=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			if((i+j)%2) s1+=min((a[i][j]%(2*d)+2*d)%(2*d),2*d-(a[i][j]%(2*d)+2*d)%(2*d)),s2+=max((a[i][j]%(2*d)+2*d)%(2*d)-d,d-(a[i][j]%(2*d)+2*d)%(2*d));
			else s2+=min((a[i][j]%(2*d)+2*d)%(2*d),2*d-(a[i][j]%(2*d)+2*d)%(2*d)),s1+=max((a[i][j]%(2*d)+2*d)%(2*d)-d,d-(a[i][j]%(2*d)+2*d)%(2*d));
		}
	if(s1<s2)
	{
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				int up=(a[i][j]%(2*d)+2*d)%(2*d);
				if(up<=d){
					if((i+j)%2) cout<<a[i][j]-up<<" ";
					else cout<<a[i][j]+d-up<<" ";
				}
				else
				{
					if((i+j)%2) cout<<a[i][j]+d*2-up<<" ";
					else cout<<a[i][j]-(up-d)<<" ";
				}
			}
			cout<<"\n";
		}
	}
	else
	{
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				int up=(a[i][j]%(2*d)+2*d)%(2*d);
				if(up<=d){
					if((i+j)%2==0) cout<<a[i][j]-up<<" ";
					else cout<<a[i][j]+d-up<<" ";
				}
				else
				{
					if((i+j)%2==0) cout<<a[i][j]+d*2-up<<" ";
					else cout<<a[i][j]-(up-d)<<" ";
				}
			}
			cout<<"\n";
		}
	}
	return 0;
} //
相关推荐
quaer2 分钟前
弗洛伊德算法--判圈法寻找环的入口O(n) O(1)
数据结构·算法
丰色木夕1 小时前
Ubuntu22.04下源码编译安装USD-Brep
c++·ubuntu·nvidia·usd-brep
高晶1 小时前
无乘法器实现LC二阶滤波器:基于移位运算的快速算法与四种精度推导
算法·设计模式·程序员
ysbfsy2 小时前
大数计算思路
算法
高晶2 小时前
主从自动分配算法:平均值乘以设备数一半的选举基准值解析
算法·设计模式·程序员
学电子她就能回来吗2 小时前
LeetCode Hot 100|普通数组专题学习笔记(零基础版)
笔记·算法·leetcode
用户老李2 小时前
反向传播计算梯度举例
算法
niyongsheng2 小时前
多智能体连续时间内路径规划:AA-SIPP(m) 的 Rust 实现与算法
算法·rust·机器人
302wanger3 小时前
DeepSeek-V4.1-Flash-1017 免费用到1024,速来
算法
lisw053 小时前
微分与积分的本质区别解析
算法·数学建模·极限编程