0800：相似 RGB 颜色（★）

力扣第 800 题

题目

RGB 颜色 "#AABBCC" 可以简写成 "#ABC" 。

• 例如，"#15c" 其实是 "#1155cc" 的简写。

现在，假如我们分别定义两个颜色 "#ABCDEF" 和 "#UVWXYZ"，则他们的相似度可以通过这个表达式 -(AB - UV)^2 - (CD - WX)^2 - (EF - YZ)^2 来计算。

那么给你一个按 "#ABCDEF" 形式定义的字符串 color 表示 RGB 颜色，请你以字符串形式，返回一个与它相似度最大且可以简写的颜色。（比如，可以表示成类似 "#XYZ" 的形式）

任何 具有相同的（最大）相似度的答案都会被视为正确答案。

示例 1：

输入：color = "#09f166"
输出："#11ee66"
解释：
因为相似度计算得出 -(0x09 - 0x11)^2 -(0xf1 - 0xee)^2 - (0x66 - 0x66)^2 = -64 -9 -0 = -73
这已经是所有可以简写的颜色中最相似的了

示例 2：

输入：color = "#4e3fe1"
输出："#5544dd"

提示：

• color.length == 7
• color[0] == '#'
• 对于任何 i > 0，color[i] 都是一个在范围 ['0', 'f'] 内的 16 进制数

分析

显然 XYZ 的取值是相互独立的。对于 AB，在 (A-1)%16,A,(A+1)%16 中找 X 使得 XX 和 AB 最近即可。

解答

1
2
3
4
5
6
7

def similarRGB(self, color: str) -> str:
	def find(pair):
		a, b = int(pair[0], 16), int(pair[1], 16)
		x = min([a,(a-1)%16,(a+1)%16], key=lambda x: abs(x*16+x-a*16-b))
		return hex(x)[2:]*2
	
	return '#'+find(color[1:3])+find(color[3:5])+find(color[5:])

28 ms