AT_abc116_d [ABC116D] Various Sushi

题目描述

现有 $N$ 个寿司。每个寿司有两个参数:“寿司种类” $t_i$ 和 “美味程度” $d_i$。您现在需要在这 $N$ 个寿司中选择吃 $K$ 个。您的 “满足感” 会被按照如下标准计算: - 满足感是 “基础美味程度总和” 和 “多样性加成” 数值的总和。 - “基础美味程度总和” 指的是你吃的所有寿司的美味程度的总和。 - “多样性加成” 是 $x \times x$,其中 $x$ 是你吃的寿司种类 (即一共有多少种 $t$)。 您现在想要得到最大的 “满足感”。找到这个 “满足感” 的最大值。

输入格式

第一行为两个整数 $N$ 和 $K$。 接下来从第 $2$ 行到第 $N + 1$ 行,第 $i$ 行两个整数 $t_i$ 和 $d_i$,分别代表第 $i$ 种寿司的寿司种类和美味程度。

输出格式

输出您可以得到的 “满足感” 的最大值。

说明/提示

- $1 \leq K \leq N \leq 10 ^ 5$ - $1 \leq t_i \leq N$ - $1 \leq d_i \leq 10 ^ 9$ - 所有输入数据均为整数 ### 样例解释 1 吃第 $1,2,3$ 个寿司时,“基础美味程度总和” 为 $9 + 7 + 6 = 22$,“多样性加成” 为 $2 \times 2 = 4$ ,得到 “满足感” 最大值为 $26$ ,可以验证不存在更好的吃法。 ### 样例解释 2 吃第 $1,2,3,4$ 个寿司,可以验证不存在更好的吃法。 ### 样例解释 3 注意数据可能会爆 $int$