CF1895F Fancy Arrays 题解

CF1895F Fancy Arrays

题意

定义一个长度为 $n$ 的非负整数序列是 Fancy 的，当且仅当：

1. $\exists i\in [1,n]$，$a_i\in [x,x+k-1]$。
2. $\forall i\in [2,n]$，$|a_i-a_{i-1}|\le k$。

$1\le n,k\le 10^9$，$0\le x\le 40$。

多测，给定 $n,x,k$，求有多少 Fancy 序列，答案对 $10^9+7$ 取模。

解析

题意就是让你求出有多少个长度为 $n$ 的数组，使得相邻两个数相差不超过 $k$，并且至少有一个数属于 $[x,x+k-1]$。

我们先看 “并且至少有一个数属于 $[x,x+k-1]$” 这一条件，需要满足这一条件，就需要同时满足 ${\max }_{i=1}^n{a}_i\ge x$ 且 ${\min }_{i=1}^n{a}_i\le x+k-1$。

用容斥原理转化一下，我们就可以，用 ${\min }_{i=1}^n{a}_i\le x+k-1$ 的方案总数减去 ${\max }_{i=1}^n{a}_i<x$ 的方案总数。

(I) 看${\min }_{i=1}^n{a}_i\le x+k-1$的方案总数如何求

我们发现只要确定了 ${\min }_{i=1}^n{a}_i$ 和 $a$ 的差分数组 $c$，就可以唯一确定数组 $a$。

因为 ${\min }_{i=1}^n{a}_i$ 要小于等于 $x+k-1$ 这个区间中所以有 $x+k$ 中方案选择。因为 $\forall c_i$ 都需要在区间 $[-k,k]$ 中 ($|c_i|\le k$)，所以有 $2k+1$ 种选择，差分数组上有 $n-1$ 个位置，所以有 $(2k+1{)}^{n-1}$ 种选择。根据简单的配列组合知识，就可以得到第一部分的总方案数是 $(x+k)\times (2k+1{)}^{n-1}$。

(II) 看${\max }_{i=1}^n{a}_i<x$ 的方案总数如何求

后者就是要求出 $\forall a_i\in [0,x)$ 的方案数，可以发现可以用 DP 求出。

令 $f_{i,j}$ 表示 $a_i=j$ 的总方案个数，可以得到转移方程 $f_{i,j}={\sum }_{t=j-k}^{j+k}{f}_{i-1,t}$，但是题目 $n$ 的数据范围太大并且 $x$ 的范围很小，又发现 $f$ 的转移式恒定不变，考虑矩阵快速幂优化，具体方法如下：

定义初始一个$1\ast x$的矩阵 $\text{F}$ 以及一个 $x\ast x$ 转移矩阵 $\text{G}$，其中 $\text{F}$的初始值都是 $1$，$\text{G}$ 的初始值若 $|i-j|\le k$ 则令 $G_{i,j}=1$， 最后求求 $n$ 次幂即可。

得到了两部分的答案，减一下就可以了，时间复杂度 $O(x^3{\log }_{2}n)$。

附代码：

#include<bits/stdc++.h>
#define INF 1e16
#define endl '\n'
#define PI acos(-1)
#define pb push_back
#define int long long
#define mem(x,v) memset(x,v,sizeof x)
using namespace std;
const int N = 45;
const int MOD = 1e9+7; 
inline int read(){
	register int x=0,f=1;register char c=getchar();
	while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
	return x*f;
}

int T,n,x,k,ans;
struct Mat{
	int a[N][N];
	friend Mat operator *(const Mat &X,const Mat &Y){
		Mat ans;mem(ans.a,0);
		for(int k=0;k<x;k++)for(int i=0;i<x;i++)for(int j=0;j<x;j++)ans.a[i][j]=(ans.a[i][j]+X.a[i][k]*Y.a[k][j])%MOD;
		return ans;
	}
}F,G;

int qpow(int x,int y){
	int ans=1;
	while(y){
		if(y&1)ans=ans*x%MOD;
		x=x*x%MOD,y>>=1;
	}
	return ans;
}
void Matqpow(int k){
	while(k){
		if(k&1)F=F*G;
		G=G*G,k>>=1;
	}
}
void solve(){
	cin>>n>>x>>k,ans=qpow(2*k+1,n-1)*(x+k)%MOD;
	for(int i=0;i<x;i++)F.a[0][i]=1;
	for(int i=0;i<x;i++)for(int j=0;j<x;j++)G.a[i][j]=(abs(i-j)<=k);
	Matqpow(n-1);
	for(int i=0;i<x;i++)ans=(ans-F.a[0][i]+MOD)%MOD;
	cout<<ans<<endl;
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	for(cin>>T;T--;)solve();
	return 0;
}