每日OJ题_牛客_空调遥控_排序+二分/滑动窗口_C++_Java

牛客_空调遥控_排序+二分/滑动窗口

描述：

dd作为集训队的队长，一直掌管着集训室的空调遥控器，她需要调整温度使队员们更好地进入训练状态，已知集训室一共有nnn名队员，每位队员都有一个温度诉求a $i$ (1≤i≤n)a $i$ (1≤i≤n)a $i$ (1≤i≤n)，当室内温度为KKK时，当且仅当∣a $i$ −K∣≤p|a $i$ -K|≤p∣a $i$ −K∣≤p时，这个队员能够正常进入训练状态，否则就会开始躁动，作为队长，dddddd需要调整好温度，她想知道，在最佳情况下，最多有多少队员同时进入训练状态。

输入描述：

第一行两个数n,p(1≤n,p≤1000000)，含义如题面描述。

接下来一行n个数a $i$ (1≤a $i$ ≤1000000)表示每个队员的温度诉求。

输出描述：

输出一个数字，表示最多有多少队员同时进入训练状态。

题目解析

先排序。

解法一：滑动窗口：维护窗口内最大值与最小值的差在 2 * p 之间。
解法二：二分查找：枚举所有的温度，二分出符合要求的学生区间，然后统计个数。

下面用滑动窗口解法：

C++代码

cpp 复制代码

#include <algorithm>
#include <iostream>
#include <vector>
#include <climits>
using namespace std;
#define int long long

signed main()
{
    // abs(e - k) <= p
    // if(e >= k) e - k <= p, e - p <= k 无语
    // if(e < k) k - e <= p, p - e >= k
    int n = 0, p = 0;
    cin >> n >> p;
    vector<int> arr(n);
    int arrMIN = INT_MAX, arrMAX = INT_MIN;
    for(int i = 0; i < n; ++i)
    {
        cin >> arr[i];
        // arrMIN = min(arrMIN, arr[i]);
        // arrMAX = max(arrMAX, arr[i]);
    }
    // sort(arr.begin(), arr.end());
    int res = INT_MIN;
    for(int k = -100; k < 100; ++k)
    {
        int cnt = 0;
        for(auto& e : arr)
        {
            if(abs(e - k) <= p)
                ++cnt;
        }
        // cout << k << " " << cnt << endl;
        res = max(res, cnt);
    }
    cout << res << endl;
    return 0;
}

Java代码

cpp 复制代码

import java.util.*;
public class Main
{
    public static void main(String[] args)
    {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt(), p = in.nextInt();
        int[] arr = new int[n];
        for(int i = 0; i < n; i++)
        {
            arr[i] = in.nextInt();
        }
        Arrays.sort(arr);

        int left = 0, right = 0, ret = 0;
        p *= 2;
        while(right < n)
        {
            while(arr[right] - arr[left] > p)
            {
                left++;
            }
            ret = Math.max(ret, right - left + 1);
            right++;
        }
        System.out.println(ret);
    }
}