第一次非AC程式碼

邏輯不夠嚴謹

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;

#define Max 100000
long long arr[Max];

int main()
{
    long long N, P;
    cin>>N>>P;
    int i;
    for( i=0; i<N; i++ )
    {
        cin>>arr[i];
    }
    sort(arr,arr+N);
    long long min=arr[0];
    for( i=N-1; i>=0; i-- )
    {
        if( arr[i]<=min*P ){
            break;
        }
    }
    cout<<i+1<<endl;
    return 0;
}
 

演算法不夠高效

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;

#define Max 100000
long long arr[Max];

int main()
{
    long long N, P;
    cin>>N>>P;
    int i;
    for( i=0; i<N; i++ )
    {
        cin>>arr[i];
    }
    sort(arr,arr+N);
    long long min=arr[0];
    for( i=N-1; i>=0; i-- )
    {
        if( arr[i]<=min*P ){
            break;
        }
    }
    cout<<i+1<<endl;
    return 0;
}
 

執行結果如下圖

AC程式碼:

#include<iostream>
#include<string>
#include<algorithm>
using namespace std;

#define Max 100000
long long arr[Max+1];

int main()
{
    long long N, P;
    int ans=1;
    cin>>N>>P;
    int i, j;
    for( i=0; i<N; i++ )
    {
        cin>>arr[i];
    }
    sort(arr,arr+N);
    for( j=0; j<N; j++ )
    for( i=j+ans; i<N; i++ )
    {
        if( arr[i]<=arr[j]*P ){
             if( i-j+1>ans )
             {
                 ans=i-j+1;
             }
        }
        else
            break;
    }
    cout<<ans<<endl;
    return 0;
}
 

後來的邏輯是 每次開始查詢的範圍隨著ans的值增大不斷縮減

換言之 就是我經過一次遍歷後 已經確定了較大的ans值 在下一次遍歷的時候我就從 j+ans 值位置開始 

類似於KMP的一步跨越儘可能多的元素個數