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$