結果

問題 No.389 ロジックパズルの組み合わせ
コンテスト
ユーザー takeya_okino
提出日時 2019-09-19 22:05:36
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 1,213 bytes
コンパイル時間 2,276 ms
コンパイル使用メモリ 77,756 KB
実行使用メモリ 75,472 KB
最終ジャッジ日時 2024-07-20 06:37:43
合計ジャッジ時間 36,591 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 47 WA * 52
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

public class Main {
  public static long MOD = (long)Math.pow(10, 9) + 7;
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    long m = sc.nextLong();
    ArrayList<Long> h = new ArrayList<Long>();
    long sum = 0;
    for(int i = 0; i < 1000000; i++) {
      try {
        long t = sc.nextLong();
        sum += t;
        h.add(t);
      } catch(Exception e) {
        break;
      }
    }
    long k = h.size();
    sum += (k - 1);
    long ans = 1;
    if(h.get(0) != 0) {
      if(sum > m) {
        System.out.println("NA");
      } else {
        long t = m - sum;
        long r1 = 1;
        long r2 = 1;
        for(long i = 1; i <= k; i++) {
          r1 = (r1 * (t + i)) % MOD;
          r2 = (r2 * i) % MOD;
        }
        long r3 = func(r2, MOD - 2);
        ans = (r1 * r3) % MOD;
        System.out.println(ans);
      }
    } else {
      System.out.println(1);      
    }
  }

  public static long func(long a, long x) {
    if(x == 1) return 1;
    if(x % 2 == 0) {
      long t = func(a, x / 2);
      return (t * t) % MOD;
    } else {
      long t = func(a, x / 2);
      t = (t * t) % MOD;
      return (a * t) % MOD;
    }
  }
}
0