牛客小白賽 B 組隊 隊列 | 區間思維

題目鏈接

題意：

給你n個數， 讓你組隊， 問如何組能組成一個小組，使這個小組元素最多且這個小組滿足裏面得元素差值兩兩等於小於K.

思路：

隊列和區間思維差不多。
先排序
再遍歷
每次更新一段新的區間， 如果出現當前差值大於k， 那麼增加左區間，反之增加右區間的值。l - r即爲小組的元素個數。

隊列

#include<bits/stdc++.h>
using namespace std;
long long a[200005];
int main()
{
    int t;
    cin >> t;
    queue<long long>q,f;
    while(t--)
    {
        int n;
        int k;
        cin >> n >> k;
        for(int i = 1;i <= n;i++)cin >> a[i];
        sort(a + 1, a + n + 1);long long ans = 0, cnt = 0;
        for(int i = 1;i < n;i++)
        {
            long long x = a[i + 1] - a[i];
            q.push(x);
            f.push(x);
        }
        long long sum = 1;
        while(!f.empty())
        {
            long long x = f.front();f.pop();
            cnt += x;
            sum++;
            if(cnt > k)
            {
                int x1 = q.front();
                q.pop();
                cnt -= x1;
                sum--;
            }
            else
            {

                ans = max(ans, sum);
            }
            ans = max(ans, sum);
        }
        ans = max(ans, sum);
        cout << ans << endl;
       while (!q.empty()) q.pop();
       while (!f.empty()) f.pop();

    }
    return 0;
}

思維

#include<bits/stdc++.h>
using namespace std;
long long a[200005];
int main()
{
    int t;
    cin >> t;
    queue<long long>q,f;
    while(t--)
    {
        int n;
        int k;
        cin >> n >> k;
        for(int i = 1;i <= n;i++)cin >> a[i];
        sort(a + 1, a + n + 1);
        int l = 1, r = 2;int ans = 0;
        while(r <= n)
        {
            if(a[r] - a[l] <= k)
            {
                r++;
            }
            else
            {
                l++;
            }
            ans = max(ans, r - l);
        }
        cout << ans << endl;
    }
    return 0;
}