给定序列a,选k个数排成一排,从左往右扫,如果当前数小于上一个数,那么当前数变成上一个数,然后形成最后的序列,问形成的序列有多少种

题目

#include
using namespace std;
#define int long long
const int maxn = 3005, mod = 1e9 + 7;
int a[maxn], b[maxn], cnt_low[maxn];//cnt_low[i]表示小于i的数字个数
int sum[maxn][maxn];//前缀和
int f[maxn][maxn];//f[i][j]表示选了i个数(不一定是原始序列的前i个),最大的数是j的方案数
									//转移:f[i][j] = f[i - 1][1] + f[i - 1][2] + ... + f[i - 1][j - 1](前i-1选的最大数小于j,那么第i个数就得选j) + f[i - 1][j](第i个数在小于等于j的数字中随便选)
void init(int n, int m){
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= m; j++){
			//f[i][j] = 1;
			f[i][j] = 0;
			sum[i][j] = 0;
		}
	}
	for(int i = 1; i <= m; i++){
		cnt_low[i] = 0;
	}
}
void solve(){
	int n, i, j;
	cin >> n;
	for(i = 1; i <= n; i++){
		cin >> a[i];
		b[i] = a[i];
	}
	sort(b + 1, b + n + 1);
	int m = unique(b + 1, b + n + 1) - b - 1;
	init(n, m);
	for(i = 1; i <= n; i++){
		a[i] = lower_bound(b + 1, b + m + 1, a[i]) - b;
		cnt_low[a[i]]++;
	}
	for(i = 1; i <= m; i++){
		cnt_low[i] += cnt_low[i - 1];
	}
	// for(i = 1; i <= n; i++){
	// 	a[i] = lower_bound(b + 1, b + m + 1, a[i]) - b;
	// 	if(!cnt_low[a[i]]) cnt_low[a[i]] = a[i];不能这样写,因为b是去重过的,a[i]的排名不能代表小于等于a[i]的个数(小于真实值)
	// 	else cnt_low[a[i]]++; 
	// }
	for(i = 1; i <= n; i++){
		for(j = 1; j <= m; j++){
			if(cnt_low[j] >= i){//小于等于j的数字要多于或等于i,f[i][j]才不为零
				if(i == 1) f[i][j] = 1;
				else f[i][j] = sum[i - 1][j] % mod;
			}
			sum[i][j] = (sum[i][j - 1] + f[i][j]) % mod;
		}
		cout << sum[i][m] << '\n';
	}
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	int T;
	cin >> T;
	while(T--){
		solve();
	}
	return 0;
}

你可能感兴趣的:(2023杭电多校,算法)