[ABC238F] Two Exams 题解

题目链接

题意

N N N 个人,有两个 1 ∼ N 1\sim N 1N 排列 P , Q P, Q P,Q,在其中选择 K K K个数,要满足:如果 P x < P y P_x<P_y Px<Py Q x < Q y Q_x<Q_y Qx<Qy 则不能选了 y y y 而不选 x x x

思路

首先按照 P P P 从小到大排序,这样的话只用考虑 Q Q Q

f i , j , k f_{i,j,k} fi,j,k 表示从前 i i i 个数中选 j j j 个,其中未选的人的 Q Q Q 值最小为 k k k

考虑第 i i i 个人选活不选:

  • 选,需要满足 Q i < k Q_i < k Qi<k,因为 P i ≥ k P_i \ge k Pik f i , j , k = f i − 1 , j − 1 , k f_{i,j,k}=f_{i-1,j-1,k} fi,j,k=fi1,j1,k
  • 不选, f i , j , min ⁡ ( k , Q i ) = f i − 1 , j , k f_{i,j,\min(k,Q_i)}=f_{i-1,j,k} fi,j,min(k,Qi)=fi1,j,k

最后答案为 ∑ i = 1 n + 1 f n , K , i \sum^{n+1}_ {i=1}{f_{n,K,i}} i=1n+1fn,K,i

时间复杂度为 O ( N 3 ) O(N^3) O(N3)

代码

/*Code by Ji-Siqi*/
/*Begin*/
#include <bits/stdc++.h>
#define mod 998244353
using namespace std;
using ll = long long;
using cint = const int;

cint N = 305;

ll f[N][N][N];

int main() {
	int n, K;
	pair <int, int> p[N];
	cin >> n >> K;
	for (int i = 1; i <= n; i++) 
		cin >> p[i].first;
	for (int i = 1; i <= n; i++) 
		cin >> p[i].second;
	sort(p + 1, p + n + 1);
	f[0][0][n + 1] = 1;
	for (int i = 0; i < n; i++) 
		for (int j = 0; j <= K; j++) 
			for (int k = 1; k <= n + 1; k++) {
				if (p[i + 1].second < k && j != K) 
					f[i + 1][j + 1][k] = (f[i + 1][j + 1][k] + f[i][j][k]) % mod;
				f[i + 1][j][min(k, p[i + 1].second)] = (f[i + 1][j][min(k, p[i + 1].second)] + f[i][j][k]) % mod;
			}
	ll ans = 0;
	for (int i = 1; i <= n + 1; i++) 
		ans = (ans + f[n][K][i]) % mod;
	cout << ans;

	return 0;
}
/*End*/
  • 25
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值