結果

問題 No.823 Many Shifts Easy
ユーザー ミドリムシミドリムシ
提出日時 2019-04-26 22:14:52
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,561 bytes
コンパイル時間 1,333 ms
コンパイル使用メモリ 76,272 KB
実行使用メモリ 20,268 KB
最終ジャッジ日時 2023-08-16 14:44:33
合計ジャッジ時間 1,821 ms
ジャッジサーバーID
(参考情報)
judge15 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 18 ms
18,324 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 17 ms
18,628 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <cassert>
#include <queue>
using namespace std;
using lint = long long;
const lint mod = 1e9 + 7;
#define all(x) (x).begin(), (x).end()
#define bitcount(n) __builtin_popcountl((lint)(n))
#define fcout cout << fixed << setprecision(15)
#define highest(x) (63 - __builtin_clzl(x))
template<class T> inline void YES(T condition){ if(condition) cout << "YES" << endl; else cout << "NO" << endl; }
template<class T> inline void Yes(T condition){ if(condition) cout << "Yes" << endl; else cout << "No" << endl; }
template<class T = string, class U = char>int character_count(T text, U character){ int ans = 0; for(U i: text){ ans += (i == character); } return ans; }
lint power(lint base, lint exponent, lint module){ if(exponent % 2){ return power(base, exponent - 1, module) * base % module; }else if(exponent){ lint root_ans = power(base, exponent / 2, module); return root_ans * root_ans % module; }else{ return 1; }}
struct position{ int y, x; }; position mv[4] = {{0, -1}, {1, 0}, {0, 1}, {-1, 0}}; // double euclidean(position first, position second){ return sqrt((second.x - first.x) * (second.x - first.x) + (second.y - first.y) * (second.y - first.y)); }
template<class T, class U> string to_string(pair<T, U> x){ return to_string(x.first) + "," + to_string(x.second); } string to_string(string x){ return x; }
template<class itr> void array_output(itr start, itr goal){ string ans; for(auto i = start; i != goal; i++) ans += to_string(*i) + " "; if(!ans.empty()) ans.pop_back(); cout << ans << endl; }
template<class itr> void cins(itr first, itr last){ for(auto i = first; i != last; i++){ cin >> (*i); } }
template<class T> T gcd(T a, T b){ if(a && b){ return gcd(min(a, b), max(a, b) % min(a, b)); }else{ return a; }} template<class T> T lcm(T a, T b){ return a / gcd(a, b) * b; }
struct combination{ vector<lint> fact, inv; combination(int sz) : fact(sz + 1), inv(sz + 1){ fact[0] = 1; for(int i = 1; i <= sz; i++){ fact[i] = fact[i - 1] * i % mod; } inv[sz] = power(fact[sz], mod - 2, mod); for(int i = sz - 1; i >= 0; i--){ inv[i] = inv[i + 1] * (i + 1) % mod; } } lint C(int p, int q) const{ if(q < 0 || p < q) return 0; return (fact[p] * inv[q] % mod * inv[p - q] % mod); } };
template<class itr> bool next_sequence(itr first, itr last, int max_bound){ itr now = last; while(now != first){ now--; (*now)++; if((*now) == max_bound){ (*now) = 0; }else{ return true; } } return false; }

combination table(1e6);

int main(){
    int N, K;
    cin >> N >> K;
    lint sum[K], end[K];
    for(int i = 1; i <= K; i++){
        sum[i - 1] = table.C(K, i) * table.fact[N - i] % mod * table.inv[N - K] % mod;
        end[i - 1] = sum[i - 1];
    }
    for(int i = 1; i < K; i++){
        sum[i - 1] = (sum[i - 1] - sum[i] + mod) % mod;
    }
    for(int i = 1; i <= K; i++){
        sum[i - 1] = (sum[i - 1] * i) % mod;
        end[i - 1] = (end[i - 1] * i) % mod;
    }
    for(int i = 1; i < K; i++){
        sum[i] = (sum[i] + sum[i - 1]) % mod;
    }
    lint ans = 0;
    for(int i = 1; i <= N; i++){
        lint ansbef = ans;
        if(i == 1){
            ans += (i * table.fact[N] % mod * table.inv[N - K] % mod + mod - end[0] % mod) % mod;
        }else if(i <= K){
            ans += (i * table.fact[N] % mod * table.inv[N - K] % mod + mod + mod - sum[i - 2] - end[i - 1] % mod) % mod;
        }else{
            ans += (i * table.fact[N] % mod * table.inv[N - K] % mod + mod + mod - sum[K - 1] % mod) % mod;
        }
    }
    cout << ans % mod << endl;
}
0