HDU-7304 2023“钉耙编程”杭电多校赛（3）Out of Control

2023“钉耙编程”中国大学生算法设计超级联赛（3）Out of Control

题目大意

有$n$个数$x_1,x_2,\dots ,x_n$，在其中选$k$个数依次放入栈中。如果当前放入栈中的数$x_i$小于栈顶的数，则向栈中放入与先前的栈顶相同的数而不是$x_i$。

对于每一个$i\in [1,n]$，求$k=i$时不同栈的数量。

有$T$组数据。

$1\le T\le 100,1\le n\le 3000$

题解

将$x_i$从小到大排序，那么一组可能的栈$s_1,s_2,\dots ,s_k$合法当且仅当$s$中仅包含$x$中的数，且对于所有的$1\le i\le k$都要满足$s_i\ge x_i$。

我们先将$x_i$离散化一下。考虑$DP$。设$f_{i,j}$表示合法的长度为$i$的满足$s_i=j$的序列$s$的数量，那么状态转移式为

$f_{i,j}=\sum _{k=1}^j{f}_{i-1,k}$

其中$j\ge x_i$。

可以用前缀和来优化$DP$。

时间复杂度为$O(\sum n^2)$。

code

#include<bits/stdc++.h>
using namespace std;
const long long mod=1e9+7;
int t,n,a[3005],num[3005];
long long sum,ans,f[3005][3005];
int main()
{
	scanf("%d",&t);
	while(t--){
		scanf("%d",&n);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			num[i]=a[i];
		}
		sort(num+1,num+n+1);
		int vt=unique(num+1,num+n+1)-num-1;
		for(int i=1;i<=n;i++){
			a[i]=lower_bound(num+1,num+vt+1,a[i])-num;
		}
		sort(a+1,a+n+1);
		for(int i=1;i<=vt;i++){
			f[1][i]=1;
		}
		printf("%d\n",vt);
		for(int i=2;i<=n;i++){
			sum=ans=0;
			for(int j=a[i-1];j<a[i];j++) sum=(sum+f[i-1][j])%mod;
			for(int j=a[i];j<=vt;j++){
				sum=(sum+f[i-1][j])%mod;
				f[i][j]=sum;
				ans=(ans+f[i][j])%mod;
			}
			printf("%lld\n",ans);
		}
	}
	return 0;
}