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;
} //
相关推荐
不会就选b2 小时前
算法日常・每日刷题--<贪心>25
数据结构·算法
蒸蒸yyyyzwd2 小时前
秋招学习笔记 day46
c++·八股
Cccp.1235 小时前
【leetcode】(七)前缀树和贪心算法
数据结构·leetcode·贪心算法
千里码aicood5 小时前
基于CART算法的图书分类系统设计与实现
数据库·算法·分类
码匠许师傅5 小时前
【C++三方组件】cpp-httplib:一个头文件起 HTTP 服务
开发语言·c++·http
做cv的小昊6 小时前
【World Model】π0.5:a Vision-Language-Action Model with Open-World Generalization
人工智能·算法·机器学习·大模型·多模态·vla·世界模型
لا معنى له6 小时前
World-In-World: World Models in a Closed-Loop World
算法
无名猿6 小时前
new/delete 与 malloc/free:为什么绝对不能混用
c++·内存管理·现代c++·踩坑记录
2601_962218618 小时前
C++中decltype关键字的实现
开发语言·c++