結果

問題 No.823 Many Shifts Easy
コンテスト
ユーザー tnakao0123
提出日時 2019-04-28 22:57:48
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,443 bytes
コンパイル時間 701 ms
コンパイル使用メモリ 84,900 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-12-21 10:30:00
合計ジャッジ時間 1,278 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:68:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   68 |   scanf("%d%d", &n, &k);
      |   ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 823.cc:  No.823 Many Shifts Easy - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 100000;
const int MOD = 1000000007;

/* typedef */

typedef long long ll;

/* global variables */

ll fracs[MAX_N + 1];

/* subroutines */

inline ll powmod(ll a, int n) {  // a^n % MOD
  ll pm = 1;
  while (n > 0) {
    if (n & 1) pm = (pm * a) % MOD;
    a = (a * a) % MOD;
    n >>= 1;
  }
  return pm;
}

inline ll inv(ll a) { return powmod(a, MOD - 2); }

inline ll npk(int n, int k) {
  if (n < 0 || k < 0 || n < k) return 0;
  return fracs[n] * inv(fracs[n - k]) % MOD;
}

inline ll nck(int n, int k) {
  if (n < 0 || k < 0 || n < k) return 0;
  return npk(n, k) * inv(fracs[k]) % MOD;
}

/* main */

int main() {
  int n, k;
  scanf("%d%d", &n, &k);

  fracs[0] = 1;
  for (int i = 1; i <= n; i++) fracs[i] = fracs[i - 1] * i % MOD;

  ll inv2 = inv(2LL);

  ll a =
    (ll)n * (n + 1) % MOD * inv2 % MOD * npk(n - 1, k) % MOD;

  ll b =
    (ll)n * (n - 1) % MOD * inv2 % MOD * nck(n - 2, k - 2) % MOD *
    fracs[k] % MOD * inv2 % MOD;

  printf("%lld\n", (a + b) % MOD);
  return 0;
}
0