题解/算法 {2813. 子序列最大优雅度}
LINK: https://leetcode.cn/problems/maximum-elegance-of-a-k-length-subsequence/description/
;
按照类型进行分类, 同一类的 从大到小排序
A1 | A2, A3 ... (A1>=A2>=A3...)
B1 | B2, B3 ...
C1 | C2, C3 ...
...
(A1>=B1>=C1...)
假如选择A1, B1, C1
则答案是(A1+B1+C1) + 3*3
, 如果选择A1,B1, B2
则答案是(A1+B1+B2) + 2*2
;
令Dif = [A1, B1, C1, ...]
从大到小排序, 令Rest = [A2,A3,..., B2,B3,..., C2,C3,..., ...]
从大到小排序;
令L = min( Dif.size(), K)
(K为题意里的参数, 即你所选序列的长度), 遍历i: [L,...,1]
, 对每个i
: Dif_sum: Dif的前i个元素; Rest_sum: Rest的前(K-i)个元素
, 更新答案Dfs_sum + Rest_sum + i*i
;
这里有个问题, 你可能认为 i
代表的是: 所选序列里 不用类型的种类个数 (从更新答案的这个式子来看 确实如此), 但是 此时这里没那么简单…
1: 当i=L
时, 确实如此 即此时你选择的序列里面 类型个数确实是i
; 这点非常重要, 即i=L
这个情况 一定是正确的!;
2: 但当i<L
时, 比如i=2, L=3
, 答案可能选择的是(A1,B1 | C2)
此时他有3个类型 可是 你是按照 2*2
来更新的 即认为他是有2种类型 这显然是错误的! (当然 也可能正确, 比如(A1,B1,A2)
就是正确的);
但重点在于 虽然2:
可能是错误的, 但这个整体是正确的 即2:
可能出现的这种错误 不影响答案!
.
分析: 比如上面这种i=2, L=3, (A1,B1 | D2)
, 此时 他一定小于(A1,B1,D1)
这种方案, 而(A1,B1,D1)
这种方案 劣于(A1,B1,C1)
即i=3
的方案, 因此 虽然i=2
会出错 但他不会影响答案! (换句话说, 答案一定包含在上述过程, 而上述过程中的错误结果 一定不会影响答案);