跳转至

LeetCode: 800. 相似 RGB 颜色

1、题目描述

RGB 颜色用十六进制来表示的话,每个大写字母都代表了某个从 0 到 f 的 16 进制数。

RGB 颜色 "#AABBCC" 可以简写成 "#ABC" 。例如,"#15c" 其实是 "#1155cc" 的简写。

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

那么给定颜色 "#ABCDEF",请你返回一个与 #ABCDEF 最相似的 7 个字符代表的颜色,并且它是可以被简写形式表达的。(比如,可以表示成类似 "#XYZ" 的形式)

示例 1:

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

注意:

  • color 是一个长度为 7 的字符串

  • color 是一个有效的 RGB 颜色:对于仍和 i > 0,color[i] 都是一个在 0 到 f 范围的 16 进制数

  • 假如答案具有相同的(最大)相似度的话,都是可以被接受的

  • 所有输入、输出都必须使用小写字母,并且输出为 7 个字符

2、解题思路

取值可能由三种,

  • 前一个
  • 当前值
  • 下一个

分别判断

class Solution:
    def similarRGB(self, color: str) -> str:
        res = "#"
        res += self.get_color(color[1:3])
        res += self.get_color(color[3:5])
        res += self.get_color(color[5:7])
        return res

    def get_color(self, c):
        num = int(c, 16)
        first = int(c[0], 16)
        pre = (first + 16 - 1) % 16
        cur = first
        nex = (first + 16 + 1) % 16

        min_value = 2 ** 31 - 1
        res = 0
        for value in [pre, cur, nex]:
            temp = (value * 16 + value - num)**2
            if temp < min_value:
                min_value = temp
                res = value
        res = hex(res * 16 + res)[2:]
        if res == '0':
            return res * 2
        return res