水色プログラミング

競プロで解いた問題の記録,ゲーム制作の進捗など...

PCK 2015 予選:4 - AZAS

問題

Aizu Online Judge

解法

愚直に O(N^2) で実装した.添え字や配列のミスがあり 2 WA ほどしてしまったのが反省.

実は(先輩に教えて頂いたが),要素ごとに,A に属するものに +1,B に属するものに +2,C に属するものに +4 すると,要素ごとの合計が 4, 6 のものが当てはまる.このようにすると凄いコードが短く簡潔に済むが,思いつかなかった.

解答

Aizu Online Judge

f:id:babcs2035:20180823192721p:plain

gist.github.com