Codeforces 135E Weak Subsequence(组合计数)

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;


#define N 1000030
#define mod 1000000007


int qpow(int x, int k) {
	int ret = 1;
	while(k) {
		if(k & 1) ret = 1LL * ret * x % mod;
		k >>= 1;
		x = 1LL * x * x % mod;
	}
	return ret;
}

int f[N], nf[N];
int inv[N];
int k, w;

void init() {
	nf[0] = f[0] = 1;
	for(int i = 1; i < N; ++i) {
		f[i] = 1LL * f[i-1] * i % mod;
	}
	nf[N-1] = qpow(f[N-1], mod - 2);
	for(int i = N - 2; i >= 0; --i) nf[i] = 1LL * nf[i+1] * (i + 1) % mod;

	inv[1] = 1;
	for(int i = 2; i < N; ++i) {
		inv[i] = mod - 1LL * (mod / i) * inv[mod % i] % mod;
	}
}

int main() {
	scanf("%d%d", &k, &w);
	if(k == 1) {
		puts("1");
		return 0;
	}
	init();

	int ans = 0;
	for(int t = 1; t <= k; ++t) {
		int tmp = 0;
		if(w - 1 >= t) {
			tmp = 1LL * f[k] * f[k] % mod * t % mod * qpow(k, w - t - 1) % mod * nf[k-t] % mod * nf[k-t] % mod;
		}
		else {
			tmp = 1LL * f[k] * w % mod * f[k-t+w-1] % mod * nf[k-t] % mod * nf[k-t] % mod;
		}
		ans += tmp; if(ans >= mod) ans -= mod;
	}
	ans = (ans + ans) % mod;
	for(int t = 1; t <= w - 2 && t <= k; ++t) {
		int tmp = 1LL * f[k] * f[k] % mod * t % mod * t % mod * nf[k-t] % mod * nf[k-t] % mod * qpow(k, w - t -2) % mod;
		ans -= tmp;
		if(ans < 0) ans += mod;
	}
	if(k >= w) {
		ans -= 1LL * f[k] * nf[k-w] % mod * f[k-1] % mod * nf[k-w] % mod;
		if(ans < 0) ans += mod;
		ans -= 1LL * f[k] * nf[k-w] % mod * (w - 1) % mod * (w - 1) % mod * f[k-2] % mod * nf[k-w] % mod;
		if(ans < 0) ans += mod;
	}
	if(k > 1 && k >= w - 1) {
		ans -= 1LL * k * (w - 1) % mod * f[k-1] % mod * nf[k-w+1] % mod * (w - 1) % mod * f[k-1] % mod * nf[k-w+1] % mod;
		if(ans < 0) ans += mod;
	}
	for(int t = w + 1; t <= k; ++t) {
		int tmp = 1LL * f[k] * nf[k-t] % mod * f[k-1-t+w] % mod * nf[k-t] % mod;
		tmp += 1LL * f[k] * nf[k-t] % mod * (w - 1) % mod * (w - 1) % mod * f[k-2-t+w] % mod * nf[k-t] % mod;
		if(tmp >= mod) tmp -= mod;
		ans -= tmp;
		if(ans < 0) ans += mod;
	}
	printf("%d\n", ans);
	return 0;
}