垃圾ACMer的暑假训练220716

垃圾ACMer的暑假训练220716

Codeforces

Codeforces Round #807 (Div. 2) A. Mark the Photographer

原题指路:https://codeforces.com/contest/1705/problem/A

题意

现要给编号$1\sim 2n$的$2n$个人拍照,其中第$i(1\le i\le 2n)$个人的身高为$h_i$.现$2n$个人排成两行,每行$n$个人,要求后一行的第$j(1\le j\le n)$个人至少比前一行的第$j$个人高$x$.问是否能做到.

有$t(1\le t\le 100)$组测试数据.每组测试数据第一行输入整数$n,x(1\le n\le 100,1\le x\le 1000)$.第二行输入$2n$个数$h_1,\cdots ,h_{2n}(1\le h_i\le 1000)$,分别表示每个人的身高.

对每组测试数据,若能做到,输出"YES";否则输出"NO".

思路

将$h[]$升序排列后,只需保证$h_{n+i}-h_i\ge x(1\le i\le n)$.

[证] 注意到$h[]$升序排列后,有$(n+1)$个人身高在$[h_i,h_{n+i}]$范围内.

因每行至多$n$人,则这$(n+1)$个人中至少有两人在同一列,故只需$h_{n+i}-h_i\ge x$.

代码

int main() {
	CaseT{
		int n,x; cin >> n >> x;
		vi h(2 * n);
		for (int i = 0; i < 2 * n; i++) cin >> h[i];
		sort(all(h));
		
		bool flag = true;
		for (int i = 0; i < n; i++) {
			if (h[i + n] - h[i] < x) {
				flag = false;
				break;
			}
		}
		if (flag) cout << "YES" << endl;
		else cout << "NO" << endl;
	}
}

---

---

Codeforces Round #807 (Div. 2) B. Mark the Dust Sweeper

原题指路:https://codeforces.com/contest/1705/problem/B

题意 ($1.5\mathrm{s}$)

初始时有$n$个数$a_1,\cdots ,a_n$.现可进行如下操作:选择一对下标$(i,j)s.t.a_i,\cdots ,a_{j-1}>0$,然后$a_i--,a_j++$.现需经若干次操作后能$s.t.a_1=a_2=\cdots =a_{n-1}=0$.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入整数$n(2\le n\le 2\mathrm{e}5)$,表示序列长度.第二行输入$n$个整数$a_1,\cdots ,a_n(0\le a_i\le 1\mathrm{e}9)$.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.

对每组测试数据,输出最小步数.数据保证有解.

思路I

观察样例猜测:有解的序列的前$(n-1)$中非零数的左右都是$0$.

模拟样例$02030$:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$2$	$0$	$3$	$0$

①取$(i,j)=(2,3)$,显然$a_2=2>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$1$	$1$	$3$	$0$

②取$(i,j)=(4,5)$,显然$a_4=3>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$1$	$1$	$2$	$1$

③取$(i,j)=(4,5)$,显然$a_4=2>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$1$	$1$	$1$	$2$

④取$(i,j)=(2,5)$,显然$a_2=1>0,a_3=1>0,a_4=1>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$0$	$1$	$1$	$3$

⑤取$(i,j)=(3,5)$,显然$a_3=1>0,a_4=1>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$0$	$0$	$1$	$4$

⑥取$(i,j)=(4,5)$,显然$a_4=1>0$,序列变为:

下标	$1$	$2$	$3$	$4$	$5$
数值	$0$	$0$	$0$	$0$	$5$

注意到一次操作只能改变两个数,故上述每个非零数分别处理的步数是最少的.

手模样例知:每个非零数$x$都会需$(x-1)$步变为$0$,且从第一个非零数开始到第$(n-1)$个数的位置在途中都会出现$1$,故还需$(n-idx-1)$步.

代码I

int main() {
	CaseT{
		int n; cin >> n;
		vi a(n);
		ll ans = 0;
		int idx = -1;
		for (int i = 0; i < n; i++) {
			cin >> a[i];
			if (i != n - 1 && a[i]) {
				ans += a[i] - 1;
				if (idx == -1) idx = i;
			}
		}
		if (~idx) ans += (ll)n - idx - 1;
		cout << ans << endl;

	}
}

---

思路II

删除序列的前导零,设当前$a_1\ne 0,k$为$a_1,\cdots ,a_{n-1}$中$0$的个数,则$ans=(a_1+\cdots +a_{n-1})+k$.

[证] 设先将序列中所有的$0$变为$1$,即取$i$为第一个非零数的下标,$j$为第一个$0$的下标,做一次操作.

将$a_1,\cdots ,a_{n-1}$中的所有$0$变为$1$需$k$次操作.

从左往右依次取$i$为第一个非零数的下标,$j=n$,做$(a_1+\cdots +a_{n-1})$次操作将$a_1,\cdots ,a_{n-1}$变为$0$.

下面证明这样的步数最小,只需证明每次操作至多只会让$ans$减少$1$.

①若操作的$j=n$,则它会使得$(a_1+\cdots +a_{n-1})$减少$1$.

②若操作的$j\ne n$,则它不会减少$(a_1+\cdots +a_{n-1})$,但至多会让$k$减少$1$.

代码II

int main() {
	CaseT{
		int n; cin >> n;
		vi a(n);
		for (int i = 0; i < n; i++) cin >> a[i];

		int start = 0;  // 指向序列的第一个非零数
		while (start < n && !a[start]) start++;

		ll ans = 0;
		for (int i = start; i < n - 1; i++) {
			ans += a[i];
			if (!a[i]) ans++;
		}
		cout << ans << endl;
	}
}

---

---

Codeforces Round #807 (Div. 2) C. Mark and His Unfinished Essay

原题指路:https://codeforces.com/contest/1705/problem/C

题意 ($2\mathrm{s}$)

给定一个长度为$n$的字符串$s$,下标从$1$开始.现有$c$次操作,每次取区间$[l,r]$,将$s[l\cdots r]$接到$s$的后面.操作结束后有$q$个询问,每个询问给定一个数$k$,输出当前字符串的第$k$个字符.

有$t(1\le t\le 1000)$组测试数据.每组测试数据第一行输入三个整数$n,c,q(1\le n\le 2\mathrm{e}5,1\le c\le 40,1\le q\le 1\mathrm{e}4)$.第二行输入长度为$n$的初始字符串$s$.接下来$c$行每行输入两个整数$[l,r](1\le l\le r\le 1\mathrm{e}18)$,表示要操作的区间,数据保证$r$不超过当前字符串的长度.接下来$q$行每行输入一个整数$k(1\le k\le 1\mathrm{e}18)$,表示询问的下标,数据保证$k$不超过当前字符串的长度.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$,$q$之和不超过$1\mathrm{e}4$.

对每个询问,输出当前字符串的第$k$个字符.

思路I

维护每次操作后字符串的长度的前缀和$pre[]$.对每个询问$k$,找到下标$k$在哪一段区间中,并递推找到上一段区间.时间复杂度$O(cq)$.

若每次询问都二分,时间复杂度$O((c\log c)q)$,不如直接递推.

代码I

int main() {
	CaseT{
		int n,c,q; cin >> n >> c >> q;
		string s; cin >> s;
    
		vl l(c), r(c);  // 操作区间的左右端点
		vl len(c + 1);  // 序列长度的前缀和
		len[0] = n;  // 初始长度
		for (int i = 0; i < c; i++) {
			cin >> l[i] >> r[i];
			len[i + 1] = len[i] + r[i] - l[i] + 1;
		}

		while (q--) {
			ll k; cin >> k;
			for (int i = c - 1; i >= 0; i--)  // 找到k属于第几段区间
				if (k > len[i]) k = l[i] + (k - len[i] - 1);  // 递推到上一个区间
			cout << s[k - 1] << endl;
		}
	}
}

---

思路II

对第$i$次操作,记录改变的字符串的起点$lef{t}_i$和终点$righ{t}_i$,并记录相邻两区间左端点的距离$dif{f}_i$,则对$k\in [lef{t}_i,righ{t}_i]$,第$k$个字符与第$(k-diff)$个字符相同,递推到初始字符串的区间即可.时间复杂度$O(cq)$.

代码II

int main() {
	CaseT{
		int n,c,q; cin >> n >> c >> q;
		string s; cin >> s;

		vl left(c + 1), right(c + 1);  // 操作区间的左右端点
		left[0] = 0, right[0] = n;  // 原字符串
		vl diff(c + 1);  // 相邻两区间的左端点的距离

		for (int i = 1; i <= c; i++) {
			ll l, r; cin >> l >> r;
			l--, r--;
			left[i] = right[i - 1];  // 当前区间的起点是上一个区间的终点
			right[i] = left[i] + (r - l + 1);
			diff[i] = left[i] - l;
		}

		while (q--) {
			ll k; cin >> k;
			k--;
			for (int i = c; i; i--) {
				if (k < left[i]) continue;
				else k -= diff[i];
			}
			cout << s[k] << endl;
		}
	}
}

---

思路III

DFS递归.

代码III

int main() {
	CaseT{
		int n,c,q; cin >> n >> c >> q;
		string s; cin >> s;

		vl len(c + 1);  // 序列长度的前缀和
		len[0] = n;  // 初始长度

		vll segs(c + 1);
		for (int i = 1; i <= c; i++) {
			cin >> segs[i].first >> segs[i].second;
			len[i] = len[i - 1] + segs[i].second - segs[i].first + 1;
		}

		function<char(int, ll)> dfs = [&](int pos, ll k)->char {  // 求第pos个区间的第k个字符
			if (!pos) return s[k - 1];

			ll cur = segs[pos].first - len[pos - 1] + k - 1;
			int idx = pos;
			while (idx - 1 >= 0 && len[idx - 1] >= cur) idx--;
			return dfs(idx, cur);
		};

		while (q--) {
			ll k; cin >> k;
			
			int idx = 0;
			for (int i = 0; i <= c; i++) {
				if (len[i] >= k) {
					idx = i;
					break;
				}
			}
			cout << dfs(idx, k) << endl;
		}
	}
}

---

---

Codeforces Round #807 (Div. 2) D. Mark and Lightbulbs

原题指路:https://codeforces.com/contest/1705/problem/D

题意 ($1.5\mathrm{s}$)

有一个长度为$n$的二进制串$s=s_1{s}_2\cdots s_n$.现可进行操作:取一个下标$i\in [2,n-1]s.t.s_{i-1}\ne s_{i+1}$,反转$s_i$的值.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入一个整数$n(3\le n\le 2\mathrm{e}5)$,表示二进制串的长度.第二行和第三行分别输入两长度为$n$的二进制串$s$和$t$.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.

对每组测试数据,若$s$能经若干次操作变为$t$,则输出最小步数;否则输出$-1$.

思路

显然若$s[0]\ne t[0]$或$s[n-1]\ne t[n-1]$,则无解.

对长度为$n$的二进制串$s$,构造长度为$(n-1)$的二进制串$\overline{s}=(s_1\mathrm{xor}{s}_2)(s_2\mathrm{xor}{s}_3)\cdots (s_{n-1}\mathrm{xor}{s}_n)$,则对$s$的操作即交换$\overline{s}$相邻的两bit.

以$s=000101\to t=010011$为例:

步数	$1$	$2$	$3$	$4$	$5$	$6$
$s$	$000101$	$001101$	$011101$	$011001$	$011011$	$010011$
$\overline{s}$	$00111$	$01011$	$10011$	$10101$	$10110$	$11010$

故$s$经若干次操作可变为$t$当且仅当$\overline{s}$与$\overline{t}$中的$1$的个数相等.设$\overline{s}$中$1$的位置分别为$a_1,\cdots ,a_k,\overline{t}$中$1$的位置分别为$b_1,\cdots ,b_k$,则$ans=|a_1-b_1|+|a_2-b_2|+\cdots +|a_k-b_k|$,这是因为最优策略中$\overline{s}$最左边的$1$与$\overline{t}$最左边的$1$配对,$\overline{s}$次左边的$1$与$\overline{t}$次左边的$1$配对,$\cdots$.时间复杂度$O(n)$.

代码

int main() {
	CaseT{
		int n; cin >> n;
		string a, b; cin >> a >> b;

		if (a[0] != b[0] || a[n - 1] != b[n - 1]) {
			cout << -1 << endl;
			continue;
		}

		function<vi(string)> get = [&](string s)->vi {
			vi res;
			for (int i = 0; i < n - 1; i++)
				if (s[i] != s[i + 1]) res.push_back(i);
			return res;
		};

		vi posa = get(a), posb = get(b);
		if (posa.size() != posb.size()) {
			cout << -1 << endl;
			continue;
		}

		ll ans = 0;
		for (int i = 0; i < posa.size(); i++) ans += abs(posa[i] - posb[i]);
		cout << ans << endl;
	}
}

---

---

33.1 生成函数

生成函数又称母函数,是一种形式幂级数,其每一项的系数可提供关于序列的信息.

生成函数有形式$F(x)=\sum _n{a}_n{k}_n(x)$,其中$k_n(x)$称为核函数.用$[k_n(x)]F(x)$表示$F(x)$第$n$项的核函数对应的系数,即$a_n$.

不同的核函数导出不同的生成函数,如:①普通生成函数$k_n(x)=x^n$;②指数生成函数$k_n(x)=\frac{x^n}{n!}$;③Dirichlet生成函数$k_n(x)=\frac{1}{n^x}$.

---

33.1.1 普通生成函数

有穷或无穷序列$a$的普通生成函数(Ordinary Generating Function,OGF)定义为形式幂级数$F(x)=\sum _n{a}_n{x}^n$.

若$a$有通项,则其普通生成函数的系数即通项.如:

①序列$a:123$的普通生成函数为$1+2x+3x^2$.

②序列$a:111$的普通生成函数为$\sum _{n\ge 0}{x}^n$.

③序列$a:124816$的普通生辰函数为$\sum _{n\ge 0}{2}^n{x}^n$.

设序列$a,b$的普通生成函数分别为$F(x)$和$G(x)$.

①$F(x)\pm G(x)=\sum _n(a_n+b_n)x^n$,即$F(x)\pm G(x)$是序列$(a_n+b_n)$的普通生成函数.

②$F(x)G(x)=\sum _n{x}^n\sum _{i=0}^n{a}_i{b}_{n-i}$,即$F(x)G(x)$是序列$\left(\sum _{i=0}^n{a}_i{b}_{n-i}\right)$的普通生成函数.

实际应用中不会一直保留形式幂级数的形式,而会转化为封闭形式以便更好地化简.

①序列$a:1,1,1,\cdots$的生成函数$F(x)=\sum _{n\ge 0}{x}^n$.

​ 显然$xF(x)+1=F(x)$,解得$F(x)=\frac{1}{1-x}$,这是$\sum _{n\ge 0}{x}^n$的封闭形式.

②序列$a:1,p,p^2,\cdots$的生成函数$F(x)=\sum _{n\ge 0}{p}^n{x}^n$.

​ 显然$pxF(x)+1=F(x)$,解得$F(x)=\frac{1}{1-px}$.等比级数的展开形式和封闭形式是常用的变换手段.

③序列$a:0,1,1,1,\cdots$的生成函数$F(x)=\sum _{n\ge 1}{x}^n=\frac{x}{1-x}$.

④序列$a:1,0,1,0,1,\cdots$的生成函数$F(x)=\sum _{n\ge 0}{x}^{2n}=\sum _{n\ge 0}(x^2{)}^n=\frac{1}{1-x^2}$.

⑤序列$a:1,2,3,4,\cdots$的生成函数$F(x)=\sum _{n\ge 0}(n+1)x^n=\sum _{n\ge 1}n{x}^{n-1}=\sum _{n\ge 0}(x^n{)}^{\prime }={\left(\frac{1}{1-x}\right)}^{\prime }=\frac{1}{(1-x{)}^2}$.

⑥序列$a_n=\left(\begin{array}{c}m\\ n\end{array}\right)(n\ge 0,m=Const.)$的生成函数$F(x)=\sum _{n\ge 0}\left(\begin{array}{c}m\\ n\end{array}\right)x^n=(1+x{)}^m$.

⑦序列$a_n=\left(\begin{array}{c}n+m\\ n\end{array}\right)(n\ge 0,m=Const.)$的生成函数$F(x)=\sum _{n\ge 0}\left(\begin{array}{c}n+m\\ n\end{array}\right)x^n=\frac{1}{(1-x{)}^{m+1}}$.

[证] 数归.$m=0$时,$F(x)=\frac{1}{1-x}$.

$m>0$时,$\frac{1}{(1-x{)}^{m+1}}=\frac{1}{(1-x{)}^m}\cdot \frac{1}{1-x}=\left[\sum _{n\ge 0}\left(\begin{array}{c}n+m-1\\ n\end{array}\right)x^n\right]\left(\sum _{n\ge 0}{x}^n\right)$

​ $=\sum _{n\ge 0}{x}^n\sum _{i=0}^n\left(\begin{array}{c}m+i-1\\ i\end{array}\right)=\sum _{n\ge 0}\left(\begin{array}{c}n+m\\ n\end{array}\right)x^n$. *第三个等号先枚举$x^n$,再枚举其系数

[Fibonacci数列的生成函数]

Fibonacci数列$a_0=0,a_1=1,a_n=a_{n-1}+a_{n-2}(n\ge 1)$.

设其普通生成函数为$F(x)$,则$F(x)=\sum _{n\ge 0}{a}_n{x}^n,xF(x)=\sum _{n\ge 1}{a}_{n-1}{x}^n,x^{2}F(x)=\sum _{n\ge 1}{a}_{n-2}{x}^n$.

注意到$xF(x)+x^{2}F(x)=\sum _{n\ge 2}(a_{n-1}+a_{n-2})x^n+a_{0}x=\sum _{n\ge 2}{a}_n{x}^n+a_{0}x$,

则$F(x)=xF(x)+x^{2}F(x)-a_{0}x+a_{1}x+a_0$,解得$F(x)=\frac{x}{1-x-x^2}$,这是Fibonacci数列的一个封闭形式.

下面求其展开形式:

[解1] $F(x)=\frac{x}{1-(x+x^2)}=\sum _{n\ge 0}(x+x^2{)}^n=\sum _{n\ge 0}\sum _{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)x^{2i}{x}^{n-i}$

​ $=\sum _{n\ge 0}\sum _{i=0}^n\left(\begin{array}{c}n\\ i\end{array}\right)x^{n+i}=\sum _{n\ge 0}{x}^n\sum _{i=0}^n\left(\begin{array}{c}n-i\\ i\end{array}\right)$.

[解2] 设$\frac{x}{1-x-x^2}=\frac{A}{1-ax}+\frac{B}{1-bx}=\frac{A-Abx+B-aBx}{(1-ax)(1-bx)}$,则$\{\begin{array}{l}A+B=0\\ -Ab-aB=1\\ a+b=1\\ ab=-1\end{array}$,解得$\{\begin{array}{l}A=\frac{1}{\sqrt{5}}\\ B=-\frac{1}{\sqrt{5}}\\ C=\frac{1+\sqrt{5}}{2}\\ D=\frac{1-\sqrt{5}}{2}\end{array}$,

故$\frac{x}{1-x-x^2}=\sum _{n\ge 0}{x}^n\frac{1}{\sqrt{5}}\left[{\left(\frac{1+\sqrt{5}}{2}\right)}^n-{\left(\frac{1-\sqrt{5}}{2}\right)}^n\right]$,这是Fibonacci数列的l另一个封闭形式.

对多项式$P(x),Q(x)$,可用类似的方法求出生成函数$\frac{P(x)}{Q(x)}$的展开式.

如求$G(x)=\frac{1}{(1-x)(1-2x{)}^2}$的展开式,因$G(x)=\frac{1}{1-x}+\frac{-2}{1-2x}+\frac{2}{(1-2x{)}^2}$,

则$[x^n]G(x)=1-2^{n+1}+(n+1)\cdot 2^{n+1}$.

[Newton二项式定理]

重定义组合数$\left(\begin{array}{c}r\\ k\end{array}\right)=\frac{r^{\underline{k}}}{k!}(r\in \mathbb{C},k\in \mathbb{N})$,则对$\forall \alpha \in \mathbb{C}$,有$(1+x{)}^{\alpha }=\sum _{n\ge 0}\left(\begin{array}{c}\alpha \\ n\end{array}\right)x^n$.

二项式定理是Newton二项式定理的特例.

[Catalan数的生成函数]

Catalan数$H_0=H_1=1,H_n=\sum _{i=0}^{n-1}{H}_i{H}_{n-i-1}(n\ge 2)$,递推式有卷积的形式.

设其普通生成函数为$H(x)$,则$H(x)=\sum _{n\ge 0}{H}_n{x}^n=1+\sum _{n\ge 1}\sum _{i=0}^{n-1}{H}_i{x}^i{H}_{n-i-1}{x}^{n-i-1}x$

​ $=1+x\sum _{i\ge 0}{H}_i{x}^i\sum _{n\ge 0}{H}_n{x}^n=1+x{H}^2(x)$,解得$H(x)=\frac{1\pm \sqrt{1-4x}}{2x}=\frac{2}{1\mp \sqrt{1-4x}}$.

代入$x=0$,得到$H(x)$的常数项$H_0$,检验得$H(x)=\frac{2}{1+\sqrt{1-4x}}$满足.

故Catalan数的生成函数的封闭形式为$H(x)=\frac{1-\sqrt{1-4x}}{2x}$.

下面将其展开.先展开$\sqrt{1-4x}$.

因$\sqrt{1-4x}=(1-4x{)}^{\frac{1}{2}}=\sum _{n\ge 0}\left(\begin{array}{c}\frac{1}{2}\\ n\end{array}\right)(-4x{)}^n=1+\sum _{n\ge 1}\frac{{\left(\frac{1}{2}\right)}^{\underline{n}}}{n!}(-4x{)}^n$,

注意到${\left(\frac{1}{2}\right)}^{\underline{n}}=\frac{1}{2}\cdot \frac{-1}{2}\cdot \frac{-3}{2}\cdots \frac{-(2n-3)}{2}=\frac{(-1{)}^{n-1}(2n-3)!!}{2^n}$

​ $=\frac{(-1{)}^{n-1}(2n-2)!}{2^n(2n-2)!!}=\frac{(-1{)}^{n-1}(2n-2)!}{2^{2n-1}(n-1)!}$,

则$\sqrt{1-4x}=1+\sum _{n\ge 1}\frac{(-1{)}^{n-1}(2n-2)!}{2^{2n-1}(n-1)!n!}(-4x{)}^n=1-\sum _{n\ge 1}\frac{(2n-2)!}{(n-1)!n!}2x^n=1-\sum _{n\ge 1}\left(\begin{array}{c}2n-1\\ n\end{array}\right)\frac{2}{2n-1}{x}^n$.

故$H(x)=\frac{1}{2x}\sum _{n\ge 1}\left(\begin{array}{c}2n-1\\ n\end{array}\right)\frac{2}{2n-1}{x}^n=\sum _{n\ge 1}\left(\begin{array}{c}2n-1\\ n\end{array}\right)\frac{1}{2n-1}{x}^{n-1}$

​ $=\sum _{n\ge 0}\left(\begin{array}{c}2n+1\\ n+1\end{array}\right)\frac{1}{2n+1}{x}^n=\sum _{n\ge 0}\left(\begin{array}{c}2n\\ n\end{array}\right)\frac{1}{n+1}{x}^n$,即$H_n=\frac{1}{n+1}\left(\begin{array}{c}2n\\ n\end{array}\right)$.

---

33.1.1.1 食物

原题指路:https://darkbzoj.cc/problem/3028

题意

有如下七种食物的选择要求(假设都以"个"为单位):①汉堡:偶数个;②可乐:$0$个或$1$个;③鸡腿:$0$个、$1$个或$2$个;④蜜桃:奇数个;⑤鸡块:$4$的倍数个;⑥包子:$0$个、$1$个、$2$个或$3$个;⑦烤肉:不超过$1$个;⑧面包:$3$的倍数个.

输入整数$n(1\le n\le 1\mathrm{e}500)$,问食物选的个数之和为$n$的方案数,答案对$10007$取模.

思路

对每种食物,设$a_n$表示该种食物选$n$个的方案数,并求出其生成函数.两种或多种食物共选$n$个的方案数的生成函数是每种食物的生成函数的卷积,故只需分别求出每种食物的方案数的生成函数,将它们相乘即可.

①$\sum _{n\ge 0}{x}^{2n}=\frac{1}{1-x^2}$;②$1+x$;③$1+x+x^2=\frac{1-x^3}{1-x}$;④$\sum _{n\ge 1}{x}^{2n-1}=\frac{x}{1-x^2}$;

⑤$\sum _{n\ge 0}{x}^{4n}=\frac{1}{1-x^4}$;⑥$1+x+x^2+x^3=\frac{1-x^4}{1-x}$;⑦$1+x$;⑧$\sum _{n\ge 0}{x}^{3n}=\frac{1}{1-x^3}$.

全部相乘即答案的生成函数$F(x)=\frac{(1+x)(1-x^3)x(1-x^4)(1+x)}{(1-x^2)(1-x)(1-x^2)(1-x^4)(1-x)(1-x^3)}=\frac{x}{(1-x{)}^4}$.

注意到$\sum _{n\ge 0}\left(\begin{array}{c}n+m\\ n\end{array}\right)x^n=\frac{1}{(1-x{)}^{m+1}}$,则$\frac{x}{(1-x{)}^{m+1}}=\sum _{n\ge 0}\left(\begin{array}{c}n+m\\ n\end{array}\right)x^{n+1}=\sum _{n\ge 1}\left(\begin{array}{c}n+m-1\\ n-1\end{array}\right)x^n$.

取$m=3$,则$F(x)=\sum _{n\ge 1}\left(\begin{array}{c}n+2\\ n-1\end{array}\right)x^n$.故$ans=\left(\begin{array}{c}n+2\\ n-1\end{array}\right)=\left(\begin{array}{c}n+2\\ 3\end{array}\right)=\frac{n(n+1)(n+2)}{6}$.

代码

const int MOD = 10007;

int main() {
	function<int(void)> input = []()->int {
		int x = 0;
		int sgn = 1;
		char ch = getchar();
		while (ch < '0' || ch > '9') {
			if (ch == '-') sgn = -1;
			ch = getchar();
		}
		while (ch >= '0' && ch <= '9') {
			x = (x * 10 + ch - '0') % MOD;
			ch = getchar();
		}
		x *= sgn;
		return x;
	};

	int n = input();

	int inv6 = qpow(6, MOD - 2, MOD);
	int ans = (ll)n * (n + 1) % MOD * (n + 2) % MOD * inv6 % MOD;
	cout << ans << endl;
}

---

---