UVa 1636 - Headshot（概率）

链接：

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4511

 

题意：

首先在手枪里随机装一些子弹，然后抠了一枪，发现没有子弹。
你希望下一枪也没有子弹，是应该直接再抠一枪（输出SHOOT）呢，还是随机转一下到任意位置再抠（输出ROTATE）？
如果两种策略下没有子弹的概率相等，输出EQUAL。
手枪里的子弹可以看成一个环形序列，开枪一次以后对准下一个位置。
例如，子弹序列为0011时，第一次开枪前一定在位置1或2（因为第一枪没有子弹），因此开枪之后位于位置2或3。
如果此时开枪，有一半的概率没有子弹。序列长度为2～100。

 

分析：

直接抠一枪没子弹的概率是一个条件概率，等于子串00的个数除以00和01总数（也就是0的个数）。
转一下到任意位置再抠没子弹的概率等于0的比率。
设子串00的个数为a，0的个数为b，则两个概率分别是a/b和b/n。问题就是比较a*n和b*b。
前者大就是SHOOT，后者大就是ROTATE，否则就是EQUAL。

 

代码：

import java.io.*;
import java.util.*;

public class Main {
    static char s[] = new char[100+5];
    
    public static void main(String args[]) {
        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        
        while(cin.hasNext()) {
            s = cin.nextLine().toCharArray();
            int a = 0, b = 0, n = s.length;
            for(int i = 0; i < n; i++) {
                if(s[i] == '1') continue;
                b++;
                if(s[(i+1)%n] == '0') a++;
            }
            if(a * n > b * b) System.out.println("SHOOT");
            else if(a * n < b * b) System.out.println("ROTATE");
            else System.out.println("EQUAL");
        }
        cin.close();
    }
}