[2018 FJ 省队集训 Day6][校内训练]传统题（组合数学妙题）

Meaning

• 一个长度为 $n$ 的序列，每个位置可以被染成 $[1,m]$ 中的任一颜色
• 求所有 $m^n$ 种染色方案的最长连续同色子段长度之和
• 答案对 $p$ 取模
• $1\le n\le 300000$
• $2\le m\le 10^8$
• $0.99\times 10^9\le p\le 1.01\times 10^9$
• $p$ 为质数
• 时限 2s
• 空限 1G

Solution - Step 1

• 神仙 zzq 的题思路果然非常巧妙
• 我们知道，对于值域为非负整数的函数 $f(S)$ ，有
• $$\sum _{S}f(S)=\sum _{i\ge 1}\sum _S[f(S)\ge i]$$
• 回到问题。直觉告诉我们，这是与最值有关的计数问题
• 所以根据上面的式子进行转化可以简化问题
• $$ans=\sum _{S}f(S)=\sum _{i\ge 1}\sum _S[f(S)\ge i]$$
• 其中 $S$ 表示一种染色方案， $f(S)$ 表示染色方案 $S$ 的最长连续同色子段长度
• 容易得出 ${\sum }_S[f[S]\ge i]$ 表示最长连续同色子段长度 $\ge i$ 的方案数
• 「最大值 $\ge i$ 」还是不好处理，考虑补集转化一下
• $$\sum _S[f(S)\ge i]=m^n-\sum _S[f(S)<i]$$
• 于是
• $$ans=n{m}^n\sum _{i=1}^n\sum _S[f(S)<i]=n{m}^n-\sum _{i=0}^{n-1}\sum _S[f(S)\le i]$$
• 发现 ${\sum }_S[f[S]\le i]$ 实际上就是把长度为 $n$ 的序列分成若干段，每段长度都不超过 i i