#41785: NA 80%


1554101-0@g.puiching.edu.mo (P6A29_1300題了)

學校 : 澳門培正中學
編號 : 216811
來源 : [60.246.150.90]
最後登入時間 :
2024-10-10 18:56:38
b542. 聽到這兩個人的身高,讓十萬人都驚呆了 -- 104學年度板橋高中校內資訊學科能力競賽(二) | From: [60.246.150.90] | 發表日期 : 2024-08-27 17:23

#include <iostream>
#include <vector>
#include <algorithm>

bool findPairWithDifferenceK(std::vector<int>& arr, int k) {
    std::sort(arr.begin(), arr.end());

    int p = 0;
    int q = 1;

    while (q < arr.size()) {
        if (arr[q] - arr[p] < k) {
            q++;
        } else if (arr[q] - arr[p] > k) {
            p++;
        } else {
            return true;
        }

        if (p == q) {
            q++;
        }
    }

    return false;
}

int main() {
    int N, Q;
    std::cin >> N >> Q;

    std::vector<int> heights(N);
    for (int i = 0; i < N; ++i) {
        std::cin >> heights[i];
    }

    for (int i = 0; i < Q; ++i) {
        int K;
        std::cin >> K;

        if (findPairWithDifferenceK(heights, K)) {
            std::cout << "YES" << std::endl;
        } else {
            std::cout << "NO" << std::endl;
        }
    }

    return 0;
}

在最後一個測資點TLE。
 
ZeroJudge Forum