Problem:
古本屋(Books)

あなたの町にはJOI 古本店という老舗の古本屋があり，あなたはJOI 古本店をよく利用している．それぞれの本には基準価格が定まっており，JOI 古本店に行けばその価格で買い取ってもらえる．

JOI 古本店では，本を小説,漫画,雑誌など10 種類のジャンルに分類して扱っている．ジャンルには1から10 までの番号が付けられている．JOI 古本店には，同じジャンルの本をまとめて買い取ってもらうと高値で買い取ってくれるというサービスがある．具体的には，同じジャンルの本をまとめてT 冊買い取ってもらう場合，そのジャンルの本の一冊あたりの買取価格が基準価格よりT - 1 円高くなる．例えば，同じジャンルで基準価格100 円，120 円，150 円の本をまとめてJOI 古本店に売ったとすると，買取価格はそれぞれ102 円，122 円，152 円となる．

さて，あなたは一身上の都合で急遽引越しをすることになった．あなたはN 冊の本を持っているが，新しい住居にすべての本を持っていくことは困難なため，N 冊の本のうちK 冊をJOI 古本店に売ることにした．

課題

N 冊の本それぞれの基準価格とジャンルの番号が与えられるとき，合計買取価格の最大値を求めるプログラムを作成せよ．

制限

2 ≤ N ≤ 2000    あなたが持っている本の冊数

1 ≤ K < N    JOI 古本店に売る本の冊数

1 ≤ Ci ≤ 100000 = 105    i 番目の本の基準価格

1 ≤ Gi ≤ 10    i 番目の本のジャンルの番号

入力

標準入力から以下の入力を読み込め．

1 行目には整数N, K が空白を区切りとして書かれており，あなたの持っている本の冊数がN で，そのうちK 冊をJOI 古本店に売ることを表す．

続くN 行にはあなたの持っている本の情報が書かれている．i + 1 行目(1 ≤ i ≤ N) には，整数Ci,Giが空白を区切りとして書かれており，i 番目の本の基準価格がCi で，ジャンルの番号がGi であることを表す．

出力

標準出力に，合計買取価格の最大値を表す整数を1 行で出力せよ．

採点基準

採点用データのうち，

配点の20% 分については，N ≤ 20 を満たす．

配点の20% 分については，すべての本のジャンルは1 または2 である．

配点の10% 分については，これら2 つの条件の両方を満たす．

配点の30% 分については，これら2 つの条件の少なくとも一方を満たす．

入出力の例

入力例

7 4
14 1
13 2
12 3
14 2
8 2
16 3
11 2

出力例

60

この入力例では，2 番目，4 番目，6 番目，7 番目の4 冊の本を売ったとき，ジャンル2 の本の買取価格が1 冊あたり2 円高くなるので，これらの本の買取価格は以下のようになる．

番号

基準価格

ジャンル

買取価格

2
4
6
7

13
14
16
11

2
2
3
2

15
16
16
13

よって合計買取価格は15 + 16 + 16 + 13 = 60 円である．このとき合計買取価格は最大となる．

問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in Kotlin (judge verdict: Wrong Answer)

import java.util.*

fun main(args:Array<String>):Unit {
    val (n, k) = readLine()!!.trim().split(' ').map(String::toInt)
    val bookStack = Array(10){ Stack<Int>() }
    repeat(n){
        val (c, g) = readLine()!!.trim().split(' ').map(String::toInt)
        bookStack[g - 1].push(c)
    }
    val book = bookStack.map{it.sortedDescending().toTypedArray()}.toTypedArray()

    val bookPrice = Array(k + 1){0}
    for (b in book){
        for (i in 1 until b.size){
            b[i] += b[i - 1] + 2 * i
        }
        for (i in b.indices){
            for (bk in (0 .. (k - i - 1)).reversed()){
                if (bookPrice[bk + i + 1] < bookPrice[bk] + b[i])
                    bookPrice[bk + i + 1] = bookPrice[bk] + b[i]
            }
        }
    }
    println(bookPrice.last())
}