質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

2回答

999閲覧

AtCoder159の「D-BannedK」の問題で何が間違っているかわからない

Msk07

総合スコア26

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2020/09/16 00:51

AtCoderのD-BannedKの問題で下記コードを提出したのですが、「WA」の結果となります。
サンプルケースではあっているのですが、一部のテストケースで「WA」となるようです。
何が間違っているのかについて教えてください。
提出結果はこちらです。

Python

1""" スニペット """ 2def get_int(): 3 return int(input()) 4 5def get_ints(): 6 return list(map(int, input().split())) 7""" スニペット """ 8import math 9# 2つ選ぶ通り数を算出 nC2 10def choose2(n): 11 return math.floor(n*(n-1)/2) 12 13# インプット 14N = get_int() 15An = get_ints() 16""" 全ての要素から2つのボールを選ぶ通り数を求める """ 17uni = [0] * (N+1) 18# ユニーク数の配列を求める 19for i in range(N): 20 uni[An[i]] += 1 21sumWay = 0 22# 各数値の2通りの選択通り数を足していく 23for i in range(N): 24 sumWay += choose2(uni[i]) 25""" 全ての要素数 - 削除する要素の通り数 + 削除する要素を引いた際の通り数 を求める """ 26for i in range(N): 27 print(sumWay - choose2(uni[An[i]]) + choose2(uni[An[i]]-1))

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答2

0

ベストアンサー

テストケースの中身

AtCoderでは、テストケースの中身が(一部のコンテストを除き)公開されています。
トップページの便利リンク集から飛べます。
今回のテストケースはこれです。

WAになる原因

uni配列では、要素i+1に数字iが書かれたボールの個数が入っています。
そのため、uni[0] = 0です。(なぜなら、1 <= A[i] <= Nなので)
そして、uni[N]には数字Nが書かれたボールの個数が入っています。
しかし、18行目のfor文は0~n-1までしか回っていません。
そこを訂正するとACしました。
少し他にも書き換えた部分があるので、質問の内容とは関係ないですが、説明します。

前計算

何度もchoose2 を計算するのは、(制約が大きくなった場合は)計算量が大きくなってしまいTLEする可能性が高くなってしまいます。
今回は、0C2 ~ nC2を求めたいので、事前にcon配列にそれぞれ代入しておきました。

投稿2020/09/16 03:38

bamboo_music_ac

総合スコア33

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

Msk07

2020/09/16 14:47

bamboo_music_acさん ご回答ありがとうございます! >AtCoderでは、テストケースの中身が(一部のコンテストを除き)公開されています。 テストケースの中身が公開されているなんて知りませんでした。 大変貴重な情報をいただきありがとうございます! WAになった原因については理解できました。 N個目の要素にアクセスできていなかったのですね。 回答例や、応用問題への方法まで教えていただき大変勉強になりました。 ここまで丁寧でわかりやすいご回答をいただき感動いたしました。 ありがとうございました!!!
bamboo_music_ac

2020/09/17 04:35

競プロは、結構有志の便利なサイトとかコンテンツがいっぱいありますよ〜!
guest

0

Python

1sumWay = 0 2# 各数値の2通りの選択通り数を足していく 3for i in range(N): 4 sumWay += choose2(uni[i])

choose2(uni[N])を足し忘れてます

投稿2020/09/16 02:22

yudedako67

総合スコア2047

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

Msk07

2020/09/16 14:49

yudedako67さん ご回答ありがとうございます。 uni[N]の要素が足せていないことを理解できました。 ありがとうございました!
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問