結果
問題 | No.190 Dry Wet Moist |
ユーザー |
|
提出日時 | 2015-04-22 10:46:22 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 413 ms / 2,000 ms |
コード長 | 1,778 bytes |
コンパイル時間 | 107 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 35,400 KB |
最終ジャッジ日時 | 2024-07-05 00:08:19 |
合計ジャッジ時間 | 5,067 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 28 |
ソースコード
def find_max_moist(As):'''2N個の整数のリスト As をN個のペアにしたとき、ペアの和が 0 となる個数の最大値を求める。'''Aplus = [a for a in As if a > 0]Aminus = [-a for a in As if a < 0]Aplus.sort()Aminus.sort()lenA0 = len(As) - len(Aplus) - len(Aminus)count = lenA0 // 2p = m = 0while p < len(Aplus) and m < len(Aminus):if Aplus[p] == Aminus[m]:count += 1p += 1m += 1elif Aplus[p] > Aminus[m]:m += 1else:p += 1return countdef find_max_wet(As):'''2N個の整数のリストAsをN個のペアにしたとき、ペアの和が正となる個数の最大値を求める。正のリストと、その他のリストとに分け、絶対値の降順に並べ換えて、貪欲法でペアを作っていく。それぞれのリストにカーソルをおき、マージソートのようにカーソルを移動させていく。'''Aplus = [a for a in As if a > 0]Arest = [-a for a in As if a <= 0]Aplus.sort(reverse=True)Arest.sort(reverse=True)lenAplus = len(Aplus)lenArest = len(Arest)count = 0p = r = 0while p < lenAplus and r < lenArest:if Aplus[p] > Arest[r]:count += 1p += 1r += 1else:r += 1if p < lenAplus:count += (lenAplus - p) // 2return countdef find_max_dry(As):minusAs = [-a for a in As]return find_max_wet(minusAs)if __name__ == "__main__":N = int(input())As = list(map(int, input().split()))dry = find_max_dry(As)wet = find_max_wet(As)moist = find_max_moist(As)print(dry, wet, moist)