PAT Basic1030 完美數列
阿新 • • 發佈:2018-11-28
給定一個正整數數列,和正整數 p,設這個數列中的最大值是 M,最小值是 m,如果 M≤mp,則稱這個數列是完美數列。
現在給定引數 p 和一些正整數,請你從中選擇儘可能多的數構成一個完美數列。
輸入格式:
輸入第一行給出兩個正整數 N 和 p,其中 N(≤105)是輸入的正整數的個數,p(≤109)是給定的引數。第二行給出 N 個正整數,每個數不超過 109 。
輸出格式:
在一行中輸出最多可以選擇多少個數可以用它們組成一個完美數列。
輸入樣例:
10 8
2 3 20 4 5 1 6 7 8 9
輸出樣例:
8
先排序,然後暴力查詢最長的符合序列,但是要進行一些優化,不然依舊會超時,具體看程式碼。
借鑑了別人的優化方法,感覺挺巧妙的。
m*p可能會超過int的儲存範圍,建議用long long儲存。
以下程式碼:
#include<algorithm>
#include<cstdio>
using namespace std;
int main()
{
long long N,p;
while(scanf("%lld%lld",&N,&p)!=EOF)
{
int i,j,max= 0; //max為最長序列長度
long long ar[100005];
for(i=0;i<N;i++)
scanf("%lld",&ar[i]);
sort(ar,ar+N);
for(i=0;i<N;i++)
{
long long mp=ar[i]*p;
for(j=max+i;j<N;j++) //這裡j=max+i,直接從現在的最長長度開始查詢
{ //如果沒有符合的,那這個max就肯定是最長的了,減少了很多不必要的迴圈
if(ar[j]>mp)
break;
if(j-i+ 1>max)
max=j-i+1;
}
}
printf("%d\n",max);
}
return 0;
}