OpenJudge_1111_Blash数集

2729:Blah数集

大数学家高斯小时候偶然间发现一种有趣的自然数集合Blah，对于以a为基的集合Ba定义如下：
(1) a是集合Ba的基，且a是Ba的第一个元素；
(2)如果x在集合Ba中，则2x+1和3x+1也都在集合Ba中；
(3)没有其他元素在集合Ba中了。
现在小高斯想知道如果将集合Ba中元素按照升序排列，第N个元素会是多少？

输入包括很多行，每行输入包括两个数字，集合的基a(1<=a<=50))以及所求元素序号n(1<=n<=1000000)

对于每个输入，输出集合Ba的第n个元素值

样例输入

1 100
28 5437

样例输出

418
900585

算法分析：

维护三个数组q1,q2,q3;

取q2、q3队首元素的较小者k，加入q1，相应队列的队首位置后移，

2*k+1、3*k+1分别加入q2、q3;

直到q1中的元素个数达到n个。

实际上，q2、q3中的元素都来自于q1，只要维护two、three两个位置，表示q2中的下一个数由q1[two]*2+1得到，q3中的下一个数由q1[three]*3+1得到，这样就不需要q2、q3这两个数组了。

特殊情况的处理：q2、q3的队首元素相同。

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String line = "";
        String[] input;
        int[] result;
        while (scanner.hasNextLine()) {
            line = scanner.nextLine();
            if (line.isEmpty()){
                break;
            }
            input = line.split(" ");
            int base = Integer.parseInt(input[0]);
            int count = Integer.parseInt(input[1]);
            result = new int[count];
            result[0] = base;
            int two = 0, three = 0;
            for (int i = 1; i < count; i++) {
                int number2 = 2 * result[two] + 1;
                int number3 = 3 * result[three] + 1;
                if (number2 == number3) {
                    result[i] = number2;
                    two++;
                    three++;
                } else if (number2 < number3) {
                    result[i] = number2;
                    two++;
                } else {
                    result[i] = number3;
                    three++;
                }
            }
            System.out.println(result[count-1]);
        }
    }
}