洛谷-基础贪心

P1223 排队接水

题目描述

有 n 个人在一个水龙头前排队接水，假如每个人接水的时间为 Ti，请编程找出这 n 个人排队的一种顺序，使得 n 个人的平均等待时间最小。

一个人的等待时间不包括他的接水时间。

如果两个人接水的时间相同，编号更小的人应当排在前面。

输入格式

第一行为一个整数 n。

第二行 n 个整数，第 i 个整数 Ti 表示第 i 个人的接水时间 Ti。

输出格式

输出文件有两行，第一行为一种平均时间最短的排队顺序；第二行为这种排列方案下的平均等待时间（输出结果精确到小数点后两位）。

输入输出样例 #1

输入 #1

10 
56 12 1 99 1000 234 33 55 99 812

输出 #1

3 2 7 8 1 4 9 6 10 5
291.90

参考答案

from collections import defaultdict

n = int(input())
nums = list(map(int, input().split()))

d = defaultdict(list)
for i, v in enumerate(nums):
    d[v].append(i + 1)
    d[v].sort()

nums.sort()
tot = 0

ans = []

for i, v in enumerate(nums):
    ans.append(d[v][0])
    d[v].pop(0)
    tot += v * (n - i - 1)

for i, x in enumerate(ans):
    print(x, end='')
    if i != n - 1:
        print(' ', end = '')
print()
res = tot / n
print(f'{res:.2f}')