首页 > 代码库 > Uva 10328 逆向思维
Uva 10328 逆向思维
传送门: https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1269
这个自己想的思路不知道哪里WA了,测试的都对,找不到数据。然后看了别人的思路感觉很棒棒啊。
题意:给你一个硬币,抛掷n次,问出现连续至少k个正面向上的情况有多少种。
原题中问出现连续至少k个H的情况,很难下手。我们可以试着将问题转化一下。
设dp[i][j]表示抛掷i个硬币出现连续至多j个H的情况种数。
因为是至多,所以j的值是可以大于i的,比如投5次,至多出现8个连续的情况,因为是递推,我们只考虑这个状态是怎么从前一个状态转移过来的
实际上原题中的出现连续至少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个位置应该是什么呢。很明显应该是F。如果是H那就会出现非法状态了。那在第i-j-1之前的位置呢。无论H和F都可以,只要不出现连续的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。
综上:
dp[i][j]表示抛掷i个硬币出现连续至多j个H的情况种数
dp[0][j]=1
i<=j:dp[i][j]=dp[i-1][j]*2
i>j :i==j+1:dp[i][j]=dp[i-1][j]*2-1
else: dp[i][j]=dp[i-1][j]*2-dp[i-j-2][j]
ans=dp[n][n]-dp[n][k-1]
因为是2的指数级可以达到100,所以要大数处理:
import java.util.*;import java.io.*;import java.math.*;public class Main{ public static void main(String[] args) { BigInteger[][] dp = new BigInteger[105][105]; for(int i=0; i<105; ++i) { dp[0][i]=new BigInteger("1"); for(int j=1; j<105; ++j) { dp[j][i]=new BigInteger("0"); if(j<=i) dp[j][i]=dp[j-1][i].multiply(BigInteger.valueOf(2)); else if(j==i+1) dp[j][i]=dp[j-1][i].multiply(BigInteger.valueOf(2)).subtract(BigInteger.valueOf(1)); else dp[j][i]=dp[j-1][i].add(dp[j-1][i].subtract(dp[j-i-2][i])); } } Scanner in = new Scanner(System.in); while(in.hasNextInt()) { int a=in.nextInt(),b=in.nextInt(); System.out.println(dp[a][a].subtract(dp[a][b-1])); } }}
Uva 10328 逆向思维