PAT A1085 Perfect Sequence / B1030 完美数列 二分/双指针

给定一个正整数数列，和正整数 p，设这个数列中的最大值是 M，最小值是 m，如果 M≤mp，则称这个数列是完美数列。

现在给定参数 p 和一些正整数，请你从中选择尽可能多的数构成一个完美数列。

输入格式：

输入第一行给出两个正整数 N 和 p，其中 N（≤10​5​​）是输入的正整数的个数，p（≤10​9​​）是给定的参数。第二行给出 N 个正整数，每个数不超过 10​9​​。

输出格式：

在一行中输出最多可以选择多少个数可以用它们组成一个完美数列。

输入样例：

10 8
2 3 20 4 5 1 6 7 8 9

输出样例：

8

选出若干个数，使得这些数中最大值不超过最小值的p倍，求出满足条件的方案中可选出的最大个数。

方法一：（二分法）先将序列按升序排列，从左至右扫描，对每一个数nums[i]，在i+1~n-1中二分查找第一个大于nums[i]*p的数的位置j,则j-i即是以nums[i]为最小值的完美序列的最大长度，取所有j-i的最大值即为题目答案。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

	//在i+1~n-1中查找第一个大于x的数的位置
	static int greaterThan(long[] nums,int i,long x) {
		if (nums[nums.length-1]<=x) {//所有数都不大于x，直接返回
			return nums.length;
		}
		int left=i+1,right=nums.length-1;
		while(left<right) {
			int mid=(left+right)/2;
			if(nums[mid]<=x) {//第一个大于x的数一定在mid右边
				left=mid+1;
			}else {//mid处大于x
				right=mid;
			}
		}
		return left;//left等于right
	}
    public static void main(String[] args) throws IOException {
    	BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));

    	String[] line=reader.readLine().split(" ");
    	int N=Integer.valueOf(line[0]),p=Integer.valueOf(line[1]);
    	line=reader.readLine().split(" ");
    	long[] nums=new long[N];
    	for(int i=0;i<N;i++)
    		nums[i]=Long.valueOf(line[i]);
    	//递增排序
    	Arrays.sort(nums);
    	int ans=1;//最大长度，初值为1
    	for(int i=0;i<N;i++) {
    		long product=nums[i]*p;
    		//在i+1~n-1中查找第一个超过product的数的位置
    		ans=Math.max(ans, greaterThan(nums,i,product)-i);
    		if(product>nums[N-1])//乘积大于最后一个数，不需要再计算
    			break;
    	}
    	System.out.println(ans);
    }
}

方法二：（two pointers）另两个下标i、j的初值均为0，i~j为所选的序列，ans记录满足条件的最大长度。让j不断增加直到nums[j]<=nums[i]*p恰好不成立，则此时序列i~j-1即为以nums[i]为最小值的最长的完美序列。然后i右移一位，j重复上述操作，直到到达序列末尾，始终让i和j的距离保持最大。时间复杂度为O(n)。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader reader=new BufferedReader(new InputStreamReader(System.in));

    	String[] line=reader.readLine().split(" ");
    	int N=Integer.valueOf(line[0]),p=Integer.valueOf(line[1]);
    	line=reader.readLine().split(" ");
    	long[] nums=new long[N];
    	for(int i=0;i<N;i++)
    		nums[i]=Long.valueOf(line[i]);
    	//递增排序
    	Arrays.sort(nums);
    	int i=0,j=0,ans=0;
    	while(j<nums.length) {
    		//j不断右移，直到恰好不满足条件
    		while(j<nums.length&&nums[j]<=nums[i]*p) {
    			j++;
    		}
    		ans=Math.max(ans, j-i);//更新最大长度
    		i++;
    	}
    	System.out.println(ans);
	}
}