結果

問題 No.1360 [Zelkova 4th Tune] 協和音
コンテスト
ユーザー ygd.
提出日時 2021-06-16 23:36:01
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 347 ms / 2,000 ms
コード長 770 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 138 ms
コンパイル使用メモリ 85,504 KB
実行使用メモリ 80,768 KB
最終ジャッジ日時 2026-06-05 02:40:05
合計ジャッジ時間 9,876 ms
ジャッジサーバーID
(参考情報)
judge1_0 / judge2_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 47
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

from math import inf


def main():
    N = int(input()); INF = float("inf")
    A = list(map(int,input().split()))
    B = [list(map(int,input().split())) for _ in range(N)]

    mx = -INF
    idx = -1

    for i in range(1,1<<N): #1個以上鳴らすので1から
        score = 0
        for j in range(N):
            if (i>>j)&1 == 1:
                score += A[j]
        for p in range(N):
            for q in range(p+1,N):
                if (i>>p)&1 == 1 and (i>>q)&1 == 1:
                    score += B[p][q]
        if score > mx:
            mx = score
            idx = i
    print(mx)
    ret = []
    for i in range(N):
        if (idx>>i)&1 == 1:
            ret.append(i+1) #1-index
    print(*ret)
    
        

if __name__ == '__main__':
    main()
0