Codeforces Round #419 (Div. 2) B. Karen and Coffee【前綴和求區間覆蓋次數】

B. Karen and Coffee

題意：

有 n 個專家給你推薦咖啡合適溫度範圍 [l,r] ，當有 k 個專家推薦到某個溫度時，說明這個溫度是合理的。
給你 q 的提問，每個提問給你一個範圍 [a,b] ，問你這個範圍有多少溫度是合理的。
數據範圍：1≤k≤n≤200000 ，1≤q≤200000 ， 1≤l≤r≤200000 ， 1≤a≤b≤200000

思路：

1.看到數據範圍這麼大，一開始我是拒絕的，後來纔想到用前綴和處理區間覆蓋問題。
2.用數組標記左右界，統計前綴和可以得到每個溫度被區間覆蓋的次數。
3.當覆蓋次數大於 k 的時候，說明這個溫度是合理的，標記爲1，否則爲0。
4.再統計一次前綴和，就可以得到 [0,i] 合理溫度的個數。
5.接下來每次詢問你懂的。

代碼：

#include <bits/stdc++.h>
using namespace std;
int a[200200];
int sum[200200];
int main() {
    int n, k, q;
    int x, y;
    scanf("%d%d%d", &n, &k, &q);
    memset(a, 0, sizeof(a));
    for(int i = 0; i < n; i++) {
        scanf("%d%d", &x, &y);
        a[x]++;
        a[y + 1]--;
    }
    int tot = 0;
    //統計前綴和，當前綴和>=k時，說明該溫度滿足條件，記該溫度爲1
    for(int i = 0; i < 200100; i++) {
        tot += a[i];
        a[i] = (tot >= k ? 1 : 0);
    }
    //再統計一下前綴和，就可以在O(1)時間內查詢一段區間滿足條件的個數了
    sum[0] = a[0];
    for(int i = 1; i < 200100; i++) {
        sum[i] = sum[i-1] + a[i];
    }
    while(q--) {
        scanf("%d%d", &x, &y);
        cout << sum[y] - sum[x-1] << endl;
    }
}