AtCoder が提供しているABC(AtCoder Beginner Contest)363 B問題をC++とPythonで解いてみました。ABC363は、2024年7月20日21:00に実施されました。
AtCoder の紹介はこちらに、プログラミングの方針はこちらに記事があります。
B問題 Japanese Cursed Doll(Difficulty : 32)
問題の詳細は、リンク先をご覧ください。
ABC363 B問題 Japanese Cursed Doll
P番目に大きい値を求めます。AtCoder Problems による Difficulty は 32 でした。
解答案
C++ プログラム例(ABC363B)
降順にソートして、$P$ 番目に大きい値を求めます。この値を $T$ から引いた値を出力します。これが負の値になれば、現時点で、髪の長さが $T$ 以上の人が $P$ 人以上にいることになるため、0 を出力します。
以下が、C++プログラムです。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, t, p;
cin >> n >> t >> p;
vector<int> L(n);
for (int i = 0; i < n; ++i) {
cin >> L[i];
}
sort(L.begin(), L.end(), greater<int>());
cout << max(0, t - L[p - 1]) << endl;
return 0;
}
AC(Accepted=正しいプログラム)と判定されました。
Python プログラム例(ABC363B)
Python版も基本的な考え方は同じです。以下がプログラムです。
"""AtCoder Beginner Contest 363 B"""
n, t, p = map(int, input().split())
L = list(map(int, input().split()))
L.sort(reverse=True)
print(max(0, t - L[p - 1]))
こちらも「AC」と判定されました。
最後に
この問題は、地道にシミュレーションしても、時間的に間に合うようです。
引き続き ABC の問題を紹介していきます。