[AtCoder] キーエンス プログラミング コンテスト 2020 C – Subarray Sum

問題

方針

\( 0 \leq K \leq N \) という制約なので、この問題は簡単に解くことができます。\( A \) の \( K \) 個の要素が \( S \) を取るとき、\( N – K \) 個の要素をどのように決めれば良いかですが、\( S = 10^9 \) のとき、\( N – K \) 個の部分列の和が \( 10^9 \) にならなければよいので、\( 1 \) としておきます。一方で、\( S \neq 10^9 \) のとき、\( N – K \) 個の値を \( S + 1 \) とすれば良いです。

コード