分數計算 / 教科書般的褻瀆

題目連結：ybt金牌導航8-3-3 / luogu P4593

題目大意

有一些怪，血量從 1~n，其中有 m 個數是沒有怪的，給出這些數。
然後你可以每次操作攻擊所有怪扣 1 血，如果有怪死了就繼續攻擊。
然後每次攻擊一個怪的分數是它當前血量的 q 次方。（q 是把所有怪打死要操作的次數）
然後問你打死所有怪的總分數。

思路

首先你要確定 \(q\)。

那你除了沒有的位置要自己打，它會一直連下去，所以 \(q=m+1\)（加一是一開始自己要打一下）

然後問你就變成求這個式子：
\(\sum\limits_{i=1}^m\sum\limits_{j=i+1}^{m+1}S(a_j-a_i-1)-S(a_{j-1}-a_i)\)

相當於幾個字首和。

然後其實第二個 \(\sum\) 還可以再弄弄：
\(\sum\limits_{i=1}^m(S(n-a_i)-\sum\limits_{j=i+1}^m(a_j-a_i)^{m+1})\)
（其實就是直接整個的字首和，然後減去中間會斷開的部分）

然後我們就可以 \(O(m^2)\) 解決了。

程式碼

#include<cstdio>
#include<algorithm>
#define ll long long
#define mo 1000000007

using namespace std;

int T;
ll n, m, ans, a[55];
ll jc[55], y[55], pre[55], suf[55];

ll ksm(ll x, ll y) {
	ll re = 1;
	while (y) {
		if (y & 1) re = re * x % mo;
		x = x * x % mo;
		y >>= 1;
	}
	return re;
}

ll Query(ll n, ll k) {//拉格朗日插值
	pre[0] = 1;
	for (int i = 1; i <= k; i++)
		pre[i] = pre[i - 1] * (n - i + mo) % mo;
	suf[k + 1] = 1;
	for (int i = k; i >= 1; i--)
		suf[i] = suf[i + 1] * (n - i + mo) % mo;
	
	ll re = 0;
	for (int i = 1; i <= k; i++) {
		re = (re + y[i] * pre[i - 1] % mo * suf[i + 1] % mo * ksm(jc[i - 1] * jc[k - i] % mo * (((k - i) & 1) ? mo - 1 : 1) % mo, mo - 2) % mo) % mo;
	}
	return re;
}

int main() {
	jc[0] = 1; for (int i = 1; i <= 54; i++) jc[i] = jc[i - 1] * i % mo;
	
	scanf("%d", &T);
	while (T--) {
		scanf("%lld %lld", &n, &m);
		for (int i = 1; i <= m; i++) scanf("%lld", &a[i]);
		sort(a + 1, a + m + 1);
		
		for (int i = 1; i <= m + 3; i++) {
			y[i] = (y[i - 1] + ksm(i, m + 1)) % mo;
		}
		ans = 0;
		for (int i = 0; i <= m; i++) {
			ans = (ans + Query(n - a[i], m + 3)) % mo;
			for (int j = i + 1; j <= m; j++) {
				ans = (ans - ksm(a[j] - a[i], m + 1) + mo) % mo;
			}
		}
		printf("%lld\n", ans);
	}
	
	return 0;
}