zoukankan      html  css  js  c++  java
  • CF908D New Year and Arbitrary Arrangement

    一道期望dp,推方程不是很难,计算边界才是本题的难点。

    我们(A)表示选(a)的概率,(B)表示选(b)的概率。

    我们由于选了(a)的个数与后面有极大关系,同时正序状态无穷无尽,所以我们考虑倒推回来,因此我们设(dp_{i,j})表示已经选了(i)(a)(j)(ab)了后面选(ab)的个数的期望。

    很显然有dp式:

    (dp_{i,j} = dp_{i+1,j} imes A + dp_{i,i+j} imes B)

    现在考虑一波边界,当(j leq k)(k leq i + j)时,(dp_{i,j}=i+j+ 0 imes B + 1 imes A imes B + 2 imes A^2 imes B + 3 imes A^3 imes B dots)

    使用一波我不会的错位相减法可以求出(dp_{i,j}=i+j+frac{pa}{pb})

    最后是答案,考虑到开头是(b)时不但可以自我无限转移,甚至还对答案无影响,所以我们考虑第一个字母直接为(a),所以答案为(dp_{1,0})

    代码:

    #include <cstdio>
    #include <algorithm>
    using namespace std;
    typedef long long LL;
    const LL mod = 1e9 + 7;
    const int MAXN = 1005;
    LL dp[MAXN << 2][MAXN << 2];
    LL qpow(LL a , LL b) {
    	LL res = 1;
    	while(b) {
    		if(b & 1) res = res * a % mod;
    		a = a * a % mod;
    		b >>= 1;
    	}
    	return res;
    }
    int main() {
    	LL k , pa , pb;
    	scanf("%lld %lld %lld" , &k , &pa , &pb);
    	LL A = pa * qpow(pa + pb , mod - 2) % mod;
    	LL B = pb * qpow(pa + pb , mod - 2) % mod;
    	LL C = pa * qpow(pb , mod - 2) % mod;
    	for (int i = 1; i <= k * 2; ++i) {
    		for (int j = 0; j <= k * 2; ++j) {
    			if(i + j >= k) dp[i][j] = (i + j + C) % mod;
    		}
    	}
    	for (int i = k; i >= 1; --i) {
    		for (int j = k; j >= 0; --j) {
    //			if(i + j >= k) dp[i][j] = (i + j + C) % mod;
    			if(i + j >= k) continue;
    			dp[i][j] = (A * dp[i + 1][j] % mod + B * dp[i][i + j] % mod) % mod;
    		}
    	}
    	printf("%lld" , dp[1][0]);
    	return 0;
    }
    
  • 相关阅读:
    感悟优化——Netty对JDK缓冲区的内存池零拷贝改造
    由浅入深理解Java线程池及线程池的如何使用
    Http学习笔记
    zookeeper集群配置详细教程
    kafka学习笔记——基本概念与安装
    干货——详解Java中的关键字
    Java基础巩固——排序
    你可以这么理解五种I/O模型
    Java中的NIO基础知识
    Java基础巩固——异常
  • 原文地址:https://www.cnblogs.com/Reanap/p/13429775.html
Copyright © 2011-2022 走看看