Submission #994252


Source Code Expand

#include <iostream>
#include <fstream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <algorithm>
using namespace std;
 
#define REP(i,n) for(int i=0; i<n; ++i)
#define FOR(i,a,b) for(int i=a; i<=b; ++i)
#define FORR(i,a,b) for (int i=a; i>=b; --i)
#define pi M_PI
 
typedef long long ll;
typedef vector<int> VI;
typedef vector<ll> VL;
typedef vector<VI> VVI;
typedef pair<int,int> P;
typedef pair<ll,ll> PL;

#define mod 1000000007

ll add(ll x, ll y){
    return (x+y)%mod;
}

ll sub(ll x, ll y){
    return (x+mod-y) % mod;
}

ll mul(ll x, ll y){
    return (x%mod)*(y%mod)%mod;
}

ll powll(ll x, ll y){
    ll res = 1LL;
    while(y){
        if (y & 1LL)
            res *= x;
        res %= mod;
        x = (x*x) % mod;
        y >>= 1LL;
    }
    return res;
}

ll div(ll x, ll y){
    return (x * powll(y,mod-2)) % mod;
}

ll nPr(ll n, ll r){
    ll res = 1LL;
    FOR(i,1,r){
        res = (res * ((n-i+1) % mod)) % mod;
    }
    return res;
}

ll nCr(ll n, ll r){
    ll res = nPr(n,r);
    FOR(i,1,r){
        res = (res * powll(i,mod-2)) % mod;
    }
    return res;
}

int main() {
    ll n, m;
    cin >> n >> m;
    if (m < n){
        cout << 0 << endl;
        return 0;
    }
    VL p(n);
    REP(i,n){
        p[i] = mul(powll(n-i,n-1),nCr(n-1,i));
    }
    ll ans = 0;
    REP(k,m-n+1){
        ll x = 0;
        REP(i,n){
            if (i % 2 == 0) x = add(x,p[i]);
            else x = sub(x,p[i]);
        }
        x = mul(x, powll(n-1,m-n-k));
        ans = add(ans, x);

        REP(i,n){
            p[i] = mul(p[i], n-i);
        }
    }

    cout << -1 << endl;

    return 0;
}

Submission Info

Submission Time
Task F - Road of the King
User TangentDay
Language C++14 (GCC 5.4.1)
Score 0
Code Size 1829 Byte
Status WA
Exec Time 10 ms
Memory 384 KB

Judge Result

Set Name sample all
Score / Max Score 0 / 0 0 / 1000
Status
AC × 1
WA × 2
AC × 3
WA × 10
Set Name Test Cases
sample sample-01.txt, sample-02.txt, sample-03.txt
all sample-01.txt, sample-02.txt, sample-03.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt
Case Name Status Exec Time Memory
01-01.txt AC 3 ms 256 KB
01-02.txt WA 3 ms 256 KB
01-03.txt WA 3 ms 256 KB
01-04.txt AC 2 ms 256 KB
01-05.txt WA 3 ms 256 KB
01-06.txt WA 3 ms 256 KB
01-07.txt WA 10 ms 256 KB
01-08.txt WA 10 ms 256 KB
01-09.txt WA 10 ms 384 KB
01-10.txt WA 10 ms 256 KB
sample-01.txt WA 2 ms 256 KB
sample-02.txt WA 5 ms 256 KB
sample-03.txt AC 3 ms 256 KB