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;
} //
相关推荐
坤坤子吖1 天前
Linux网络编程:TCP服务器的多进程与多线程实现
linux·服务器·开发语言·网络·c++·笔记·学习
一晌小贪欢1 天前
Python基础:第 09 篇 · while 循环:重复直到不满足
开发语言·c++·python·数据可视化·面向对象·python办公
zh_xuan1 天前
LeeCode 496. 下一个更大元素I
数据结构·算法·栈·单调栈
机 _ 长1 天前
05 智算运维|8 张卡怎么才算「捆成一张」:NCCL 实测与拓扑
运维·人工智能·算法·智算运维
xiaoye-duck1 天前
《C++ 项目:手写大模型接入 SDK》LLMManager架构设计与实现
开发语言·网络·c++·语言模型
一条大祥脚1 天前
CuTe 与 CuTe DSL:从 C++ 模板元编程到 Python DSL 的 GPU 编程演进
开发语言·c++·python·cutlass·cute·cute dsl
·present·1 天前
CSAPP(lecture 13_02 Linking)
c语言·c++
早起CaiCai1 天前
【2023-01 OM】利用深度学习算法融合异质观测实现海洋场降尺度
人工智能·深度学习·算法
梦里寻码1 天前
我用 NestJS + Vue3 写了一套完整的研发管理系统(58 表 / 229 API / 130 项断言,全开源
前端·后端·算法
旖旎夜光1 天前
【C++ 基础入门】C++ 多态(上):虚函数、重写、override/final 与抽象类
开发语言·c++·笔记·学习