ABC128 B - Guidebook

備忘録

問題

atcoder.jp

回答

import sys
import os
import math

ii = lambda: int(sys.stdin.buffer.readline().rstrip())
il = lambda: list(map(int, sys.stdin.buffer.readline().split()))
fl = lambda: list(map(float, sys.stdin.buffer.readline().split()))
iln = lambda n: [int(sys.stdin.buffer.readline().rstrip()) for _ in range(n)]

iss = lambda: sys.stdin.buffer.readline().decode().rstrip()
sl = lambda: list(map(str, sys.stdin.buffer.readline().decode().split()))
isn = lambda n: [sys.stdin.buffer.readline().decode().rstrip() for _ in range(n)]

MOD = 10 ** 9 + 7


def main():
    if os.getenv("LOCAL"):
        sys.stdin = open("input.txt", "r")

    N = ii()
    ret = []
    for n in range(N):
        S, P = sl()
        ret.append([S, int(P), n + 1])
    ret.sort(key=lambda List: (List[0], -List[1]))
    for s, p, n in ret: print(n)


if __name__ == '__main__':
    main()

考え方

レストランの紹介順は市名が辞書順で早い場所・点数は高いほうから、と条件が決まっている。
つまり、市名は昇順・点数は降順でソートを行う必要がある。

複数条件のソートは下記を参考にした。
[Python] 多次元リストを複数キーでソートする - Qiita

市名と点数でソートを行う前に、最初の位置を持った配列を作る。
ソート後に配列を頭から順に、はじめの位置をprintすることで回答。