oj.uz

답안 #15730

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
15730 2015-07-17T02:31:39 Z gs14004 접미사 배열의 개수 (kriii3_W) C++14
0 / 46
 0 ms 8896 KB 
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
typedef long long lint;
const int mod = 1e9 + 7;

lint p(lint a, int b){
	lint ret = 1, piv = a;
	while(b){
		if(b&1) ret *= piv;
		piv *= piv;
		ret %= mod;
		piv %= mod;
		b >>= 1;
	}
	return ret;
}

lint fac[1000005];

int main(){
	int n, m;
	scanf("%d %d",&n,&m);
	m = min(n, m);
	fac[0] = 1;
	for(int i=1; i<=n; i++){
		fac[i] = (1ll * fac[i-1] * i) % mod;
	}
	lint ret = 0;
	for(int i=0; i<m; i++){
		lint comb = ((1ll * fac[n] * p(fac[i], mod - 2) % mod)  * p(fac[n-i], mod - 2) % mod);
		comb *= p(-1, i) * p(m - i, n) % mod;
		ret += comb;
		ret %= mod;
	}
	printf("%lld",ret);
}

Subtask #1
0.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 8896 KB Output is correct
2 Correct 0 ms 8896 KB Output is correct
3 Correct 0 ms 8896 KB Output is correct
4 Incorrect 0 ms 8896 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #2
0.0 / 36.0

# 결과 실행 시간 메모리 Grader output
1 Halted 0 ms 0 KB -