#698611 (Python3) No.1470 Mex Sum

• 問題: No.1470 Mex Sum
• ユーザー: Mine
• 提出日時: 2021-09-10 23:26:45
• 言語: Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0)
• 結果: AC
• 実行時間: 85 ms / 2,000 ms
• コード長: 199 bytes
• コンパイル時間: 386 ms
• コンパイル使用メモリ: 12,672 KB
• 実行使用メモリ: 18,536 KB
• 最終ジャッジ日時: 2024-06-12 04:23:08
• 合計ジャッジ時間: 5,463 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	27 ms; 10,752 KB
testcase_01	AC	29 ms; 10,624 KB
testcase_02	AC	30 ms; 11,648 KB
testcase_03	AC	31 ms; 11,520 KB
testcase_04	AC	30 ms; 11,520 KB
testcase_05	AC	31 ms; 11,520 KB
testcase_06	AC	32 ms; 11,520 KB
testcase_07	AC	32 ms; 11,520 KB
testcase_08	AC	31 ms; 11,520 KB
testcase_09	AC	32 ms; 11,520 KB
testcase_10	AC	33 ms; 11,520 KB
testcase_11	AC	32 ms; 11,520 KB
testcase_12	AC	80 ms; 18,220 KB
testcase_13	AC	79 ms; 18,152 KB
testcase_14	AC	79 ms; 18,160 KB
testcase_15	AC	79 ms; 18,224 KB
testcase_16	AC	77 ms; 18,140 KB
testcase_17	AC	77 ms; 17,964 KB
testcase_18	AC	76 ms; 17,928 KB
testcase_19	AC	79 ms; 18,144 KB
testcase_20	AC	82 ms; 18,104 KB
testcase_21	AC	85 ms; 18,256 KB
testcase_22	AC	85 ms; 18,424 KB
testcase_23	AC	81 ms; 18,440 KB
testcase_24	AC	82 ms; 18,404 KB
testcase_25	AC	81 ms; 18,416 KB
testcase_26	AC	81 ms; 18,408 KB
testcase_27	AC	80 ms; 18,400 KB
testcase_28	AC	81 ms; 18,428 KB
testcase_29	AC	81 ms; 18,536 KB
testcase_30	AC	79 ms; 18,408 KB
testcase_31	AC	82 ms; 18,408 KB
testcase_32	AC	81 ms; 18,416 KB
testcase_33	AC	82 ms; 18,424 KB
testcase_34	AC	82 ms; 18,428 KB
testcase_35	AC	82 ms; 18,276 KB
testcase_36	AC	80 ms; 18,432 KB
testcase_37	AC	67 ms; 13,996 KB
testcase_38	AC	67 ms; 13,996 KB
testcase_39	AC	69 ms; 13,996 KB
testcase_40	AC	68 ms; 14,124 KB
testcase_41	AC	67 ms; 13,996 KB
testcase_42	AC	67 ms; 13,996 KB
testcase_43	AC	70 ms; 14,000 KB
testcase_44	AC	68 ms; 14,000 KB
testcase_45	AC	67 ms; 14,000 KB
testcase_46	AC	70 ms; 13,996 KB
testcase_47	AC	67 ms; 13,996 KB
testcase_48	AC	68 ms; 13,996 KB
testcase_49	AC	68 ms; 14,000 KB
testcase_50	AC	68 ms; 13,996 KB

ソースコード

n = int(input())
a = list(map(int, input().split()))
one = a.count(1)
two = a.count(2)
ans = 0
ans += 3*one*two
ans += 2*one*(one-1)//2
ans += 2*one*(n-one-two)
ans += (n-one)*(n-one-1)//2
print(ans)