というDPを考えます。 ここで、答えの上限は「最もパワーの大きい色の生命体P KKK 匹で運ばせたときのパワーの総和+1+1+1」に等しいことから K×max(pi)+1K \times max(p_i)+1K×max(pi)+1 となり、 この範囲の中でDPを行うことでこのDPの時間計算量はO(NK×max(pi))O(NK \times max(p_i))O(NK×max(pi)) となります。