codeforces C. Wish I Knew How to Sort 期望DP

最新推荐文章于 2024-08-19 13:32:52 发布

黄二十九

最新推荐文章于 2024-08-19 13:32:52 发布

阅读量261

点赞数

分类专栏：算法刷题文章标签：算法

本文链接：https://blog.csdn.net/qq_30798083/article/details/127791187

版权

算法刷题专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本文解析了如何用概率论与动态规划解决Codeforces题目1753C，涉及01数组排序问题，通过抽象交换事件的概率计算期望交换次数。关键步骤包括抽象问题、构造转移方程和使用longlong避免溢出。分享了AC代码和解题思路。

摘要由CSDN通过智能技术生成

题目链接Problem - 1753C - Codeforces

题意是给你一个01数组a，每次随机选两个数字i,j(i<j)，如果a[i]>a[j]就交换两个数，问期望多少次交换能够将此序列变为一个有序的上升序列。（不要看样例，我不会求无穷级数和::>_<::）

这是一道很经典的概率DP题，首先这种题一开始一定要将问题抽象化，要将数组变得有序，就意味着将左边的1和右边的0进行交换，直到有序，假设有x个需要交换的对数，那么选择到这样一对的概率是p=x*x/Cn2。令E(x)为当前还剩x对需要交换时期望的交换次数，那么有交换成功和失败两个事件，所以E(x)=E(x-1)*p+E(x)*(1-p)+1（这个+1很关键，代表着当前交换次数+1无论失败与否）;不要想交换的顺序，这会使得问题复杂化。

AC代码

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int mod = 998244353;
int s[200010];
int dp[100010];

int qsm(int x, int n) {
	int res = 1;
	for (; n; n >>= 1, x = 1ll * x * x % mod) {
		if (n & 1)
			res = 1ll * res * x % mod;
	}
	return res;
}

int main() {
	int t;
	scanf("%d", &t);
	while (t--) {
		int n, sum = 0;
		scanf("%d", &n);
		for (int i = 1; i <= n; i++) {
			scanf("%d", &s[i]);
			if (s[i] == 1)
				sum++;
		}
		int x = sum;
		for (int i = n; i >= n - sum + 1; i--)
			if (s[i] == 1)
				x--;

		for (int i = 1; i <= x; i++) {
			dp[i] =  (dp[i - 1] +  1ll * n * (n - 1) / 2 % mod * qsm(1ll * i * i % mod, mod - 2)) % mod;
		}
		printf("%d\n", dp[x]);
	}
	return 0;
}

一定要开long long，好久没刷CF都忘了，WA了好几发