当前位置 : 主页 > 编程语言 > java >

UVA 10328 -—— Coin Toss【DP & 大数】

来源:互联网 收集:自由互联 发布时间:2022-07-13
​​题目传送门​​ 题意: 给你一个硬币,抛掷n次,问出现连续至少k个正面向上(H)的情况有多少种。 分析: 原题中问出现连续至少k个H的情况,很难下手。我们可以试着将问题转


​​题目传送门​​

UVA 10328 -—— Coin Toss【DP & 大数】_DP

题意:

给你一个硬币,抛掷n次,问出现连续至少k个正面向上(H)的情况有多少种。

分析:

原题中问出现连续至少k个H的情况,很难下手。我们可以试着将问题转化一下,设 dp[i][j] 表示抛掷i个硬币出现连续至多 j 个H 的情况种数。

实际出现连续至少k个H,即出现连续k个H,k+1个H,…n个H的并集,等价于dp[n][n]-dp[n][k-1],即从连续至多n个H的情况(所有抛掷情况种类数)减去连续至多(k-1)个H的情况,这保证得到的所有情况一定至少有k个连续的H。

现在问题就变成了怎么求dp[i][j]。

i <= j:
dp[i][j] = dp[i-1][j] × 2,即从上一阶段得到的抛掷序列后面增加正和反两种情况,如果出现连续的H个数大于j个,这种情况是非法的,但很显然此时不会出现这种情况。当前位置随便放。

i > j:
如果继续用 dp[i][j] = dp[i-1][j] × 2 就不行了。因为如果 从第 i - j 个到第 i - 1 全是H ,那么这时候在第i个位置再加一个H,就会出现连续的H个数大于j个的非法状态,所以我们需要减掉 从 i - j 到第 i - 1 全是H 的这种情况。那么这种情况有多少种呢?我们考虑该状态是如何转移而来的。试想第 i - j - 1 个位置应该是什么呢?很明显只能是T。如果是H那就会出现非法状态了。那在第 i - j - 1 之前的位置呢。无论H和T都可以,只要不出现连续的H个数大于j的非法状态即可,这就是 dp[i-j-2][j]。

那么这样,dp[i][j] = dp[i-1][j] × 2 - dp[i-j-2][j]。

但这还是不够的。我们之前的推导都是基于第i-j-1个位置一定存在的前提下(i>j不能保证第i-j-1个位置一定存在),那如果第i-j-1个位置不存在,第i-j-2个位置也就不存在,上述方程也就不成立了。但这种情况很好想,此时一定是i==j+1,从第1个位置到第j个位置全部都是H,只有这一种情况,所以方程变成 dp[i][j]=dp[i-1][j] × 2 - 1。

综上,状态转移方程为:
i<=j:dp[i][j] = dp[i-1][j] × 2;
i==j+1:dp[i][j] = dp[i-1][j] × 2 - 1;
i>j+1:dp[i][j] = dp[i-1][j] × 2 - dp[i-j-2][j]

ans=dp[n][n] - dp[n][k-1]

AC代码:

import java.math.BigInteger;
import java.util.Scanner;

public class Main {

public static void main(String[] args) {
BigInteger a, b;
BigInteger dp[][] = new BigInteger[105][105];
//初始化
for (int i = 0; i <= 100; i++) {
dp[0][i] = BigInteger.valueOf(1);
dp[i][0] = BigInteger.valueOf(1);
dp[1][i] = BigInteger.valueOf(2);
}
for (int i = 1; i <= 100; i++) {
for (int j = 1; j <= 100; j++) {
if (i <= j)
dp[i][j] = dp[i - 1][j].multiply(BigInteger.valueOf(2));
else if (i == j + 1)
dp[i][j] = dp[i - 1][j].multiply(BigInteger.valueOf(2)).subtract(BigInteger.valueOf(1));
else
dp[i][j] = dp[i - 1][j].multiply(BigInteger.valueOf(2)).subtract(dp[i - j - 2][j]);

}
}
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
a = BigInteger.valueOf(in.nextInt());
b = BigInteger.valueOf(in.nextInt()).subtract(BigInteger.valueOf(1));
System.out.println(dp[a.intValue()][a.intValue()].subtract(dp[a.intValue()][b.intValue()]));

}

in.close();
}
}


上一篇:UVA 624——CD【01背包 + 打印路径】
下一篇:没有了
网友评论