USACO 19JAN Train Tracking 2 P 题解

题目传送门

题目大意： 有一个长度为 $n$ 的序列 $a$，有一个 $k$，给出每个长度为 $k$ 的子段的最小值，问这个序列有多少种可能性。

题解

容易发现假如 $c_{i-1}>c_i$，那么就可以确定 $a_{i+k-1}=c_i$。

所以需要考虑的是一段相同的 $c_i$ 要如何处理，即考虑一个子问题：当所有 $c_i$ 相等时，这个序列有多少种可能性？

设 $f_i$ 表示长度为 $i$ 的序列的方案数，令 $d=c_1=c_2=...=c_{n-k+1},p=10^9-d$，由于每 $k$ 位至少有一个数恰好为 $c_i$，可以得到这样的 $\text{dp}$ 方程：
$$f_i=(p+1)f_{i-1}-p^k{f}_{i-k-1}$$

$(p+1)f_{i-1}$ 表示第 $i$ 位随便放，然后去掉不合法的方案，即后 $k$ 个都大于 $d$ 的方案，由于序列前 $i-1$ 位是合法的，所以当后 $k$ 个都大于 $d$ 时，第 $i-k$ 个一定为 $d$，所以方案数为 $f_{i-k-1}$。

当考虑一段相等的 $c_{[l,r]}$ 时，假如相邻的 $c$ 比它大，比如 $c_{l-1}>c_l$，那么 $a_{[l-1,l+k-2]}$ 这一段已经被前面的 $c$ 考虑了，而 $c_{l+k-1}$ 一定为 $c_l$，也不需要考虑，所以需要 $dp$ 的部分就少了 $k$ 个，去掉他们再 $dp$ 即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 100010
#define mod 1000000007

int n,k,a[maxn],ans=1;
int f[maxn];
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
int dp(int ln,int mi){
	int p=1e9-mi,p_=ksm(p,k);f[0]=1;
	for(int i=1;i<=ln;i++){
		f[i]=1ll*f[i-1]*(p+1)%mod;
		if(i>=k)f[i]=(f[i]-1ll*p_*(i>k?f[i-k-1]:1)%mod+mod)%mod;
	}
	return f[ln];
}

int main()
{
	scanf("%d %d",&n,&k);
	for(int i=1;i<=n-k+1;i++)scanf("%d",&a[i]);
	for(int i=1;i<=n-k+1;i++){
		int j=i;while(a[j+1]==a[i])j++;
		int len=j-i+k;
		if(i>1&&a[i-1]>a[i])len-=k;
		if(j<n-k+1&&a[j+1]>a[j])len-=k;
		if(len>0)ans=1ll*ans*dp(len,a[i])%mod;
		i=j;
	}
	printf("%d",ans);
}