ABC163に参加してきました!(Java使用、ネタバレ注意!)
今回もABCに参加してきたので、感想をば。結果から言うと、今回もC問題まで解いたのですが、解くまでにかかる時間が大分短くなって、19分で解けるようになりました!(今回は難易度が低かったし、サーバーの問題でunratedになったから意味ないとか言わんでくれ.....)
さて、おそらくD問題の一般解の求め方の方がA、B、C問題より需要がありそうなので(ていうか僕自身も忘れたくないので)、今回はその求め方を書いていきます。
本番中は眠すぎて、何言ってんだコイツとおm何をすればいいのかわからなかったんですが、翌日きちんと解いてみたらそんなに難しくはありませんでした。
リンク:https://atcoder.jp/contests/abc163/tasks/abc163_d
まずは、問題文。
, , ..., の 個の数があります。
この中から 個以上の数を選ぶとき、その和としてあり得るものの個数を で求めてください。
K個の数字を選んだ時に作れる和の種類を答えろとのことなので、多分各Kにおける、(最大値ー最大値)+1の値を足し合わせていけばよいのだろうと。入力例を見ても、ダブっているものは引かれていました。
入力例 1 Copy
Copy3 2出力例 1 Copy
Copy10以下の 通りが考えられます。
ちょっとはみ出てるのは許してください。一応、手で全部書き出してみたのですが、各Kに対する最小値と最大値の間の値は全種類作り出すことができます!ちなみに、異なるK内の値同士でダブルことがあるかもしれないとも一瞬頭をよぎりましたが、制約が下記の通りで、和の値の桁数高々10^10だった(ミスってたらコメください)ので、各Kごとに足し合わせていくだけで大丈夫。
制約
- 入力は全て整数
そんなこんなで書いたコードがこちら。
- import java.util.*;
- import java.util.ArrayDeque;
- import java.util.Queue;
- import java.math.RoundingMode;
- import java.math.BigDecimal;
- public class Main{
- public static void main(String[] args){
- Scanner sc = new Scanner(System.in);
- long n = sc.nextLong();
- long k = sc.nextLong();
- long mod = 1000000007L;
- long sum = 0;
- long min = 0;
- long max = 0;
- for(long i = k; i <= n+1; i++) {
- min = i*(i-1)/2;
- max = i*(2*n-(i-1))/2;
- sum+=(max-min+1)%mod;
- }
- System.out.println(sum%mod);
- }
- }
きちんと出力する時点でもう1度modの処理をしとくのもお忘れなく!
最近はこんな感じのきちんと題意を理解すれば実装はBレベルな問題が多いですな~。頭が柔らかくなるように引き続き頑張ります!