PAT Basic 完美数列(25) [two pointers]

题目

给定⼀个正整数数列,和正整数p,设这个数列中的最⼤值是M,最⼩值是m,如果M <= m * p,则称这个数列是完美数列。现在给定参数p和⼀些正整数,请你从中选择尽可能多的数构成⼀个完美数列。
输⼊格式:
输⼊第⼀⾏给出两个正整数N和p,其中N(<= 10^5)是输⼊的正整数的个数,p(<= 10^9)是给定的参数。第⼆⾏给出N个正整数,每个数不超过109。
输出格式:
在⼀⾏中输出最多可以选择多少个数可以⽤它们组成⼀个完美数列。
输⼊样例:
10 8
2 3 20 4 5 1 6 7 8 9
输出样例:
8

题目分析

已知正整数序列seq[N],最大值为M,最小值为m,已知另一个正整数p(<=10^9),从数列中抽出一部分数字,求可以满足M<=m*p的数字最多抽取个数

要满足M<=mp抽取的数字最多(即:M与m中间夹的数字最多),需要取所有满足M<=mp的情况中,m最小,M最大

解题思路

思路 01(最优、二分查找、查找M复杂度O(logn))

  1. 对seq[N]升序排序
  2. 依次遍历seq[i],在i+1到N之间,找到最大满足M<=mp的数字(即:第一个满足大于mp的数字下标j-1)

思路 02 (two pointer、查找M复杂度O(n))

  1. 对seq[N]升序排序
  2. 依次遍历seq[i],j初始为0,开始从上次j往后找(因为i+1后m增大,m*q>=M,所以M增大,j只能在上次j之后)

易错点

  1. p(<=10^9),所以m*p有可能超过int范围,数组元素类型需为long long,否则第5个测试点错误
  2. 取第一个大于mp的数字下标-1,而不是第一个大于等于mp的数字下标(因为大于的情况下要-1,等于的情况下不需要-1,处理麻烦)
  3. 思路02中,只能从前往后找第一个不满足条件m*q>=M的,不能从后往前找最后一个满足条件的(测试点4超时)

Code

Code 01

#include <iostream>
#include <algorithm>
using namespace std;
int main(int argc,char * argv[]) {
	int n,p;
	scanf("%d %d",&n,&p);
	long long seq[n]= {0}; // 若为int,第5个测试点错误 
	for(int i=0; i<n; i++) {
		scanf("%d",&seq[i]);
	}
	sort(seq,seq+n);
	int maxnum=0;
	for(int i=0; i<n; i++) {
		// 二分查找
		int left=i+1,right=n;
		int mid = left+((right-left)>>1);
		while(left<right) {
			mid = left+((right-left)>>1);
			if(seq[mid]>seq[i]*p) { //若是求第一个大于等于seq[i]*p,测试点2错误 
				right=mid;
			} else {
				left=mid+1;
			}
		}
		if(right-i>maxnum)maxnum=right-i;
	}
	printf("%d",maxnum); 
	return 0;
}

Code 01

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int main(int argc,char * argv[]) {
	int n,p;
	scanf("%d %d",&n,&p);
	long long seq[n]= {0}; // 若为int,第5个测试点错误
	for(int i=0; i<n; i++) {
		scanf("%d",&seq[i]);
	}
	sort(seq,seq+n);
	// 写法一:
	int maxnum=0,j = 0;
	for(int i=0; i<n; i++) {
		while(j<n&&seq[i]*p>=seq[j]) j++;
		maxnum=max(maxnum,j-i);
	}

	// 写法二:
//	int i=0,j=0,maxnum=1;
//	while(i<n&&j<n) {
//		while(j<n&&seq[j]<=(long long)seq[i]*p) {
//			maxnum=max(maxnum,j-i+1);
//			j++;
//		}
//		i++;
//	}

	/*
		使用下面代码,第四个测试点超时 
		j从后往前找最后一个满足条件的,测试点4超时 
	*/
//	int maxnum=0,prej=0; //prej用于记录上次j的位置,之后的j只可能比prej大,m*p>=M;i+1因为m增大了,所以M一定增大
//	for(int i=0; i<n; i++) {
//		int j = n-1;
//		while(prej<=j&&seq[i]*p<seq[j]) j--;
//		maxnum=max(maxnum,j-i+1);
//		prej=j;
//	}

	printf("%d",maxnum);
	return 0;
}
原文地址:https://www.cnblogs.com/houzm/p/12257297.html