CCF CSP 2022.3 第二题 出行计划

题目描述

题目分析

我刚开始写的代码超时了，原因是使用了两层for循环，时间复杂度为O(n*m)，导致超时

具体思想如下,关键判断条件

需持有时间∈[查询时间+ 等待时间, 查询时间+等待时间+ 场所所需核酸结果时间 - 1]
q[i] + k <= t[j] && q[i] + k + c[j] - 1 >= t[j]

超时代码 70 分

#include <bits/stdc++.h>
​
using namespace std;
​
int main() {
    //n,m,k分别表示出行计划数目、查询个数以及等待核酸检测结果所需时间
    int n = 0, m = 0, k = 0;
    cin >> n >> m >> k;
    //用数组记录每一项的出行时间、所需核酸时间单位、查询个数的时刻
    int t[n + 1];
    int c[n + 1];
    int q[m + 1];
​
    for (int i = 0; i < n ; i++) { //将每一项出行计划的时刻和所需核酸单位进行保存
        int a = 0, b = 0;
        cin >> a >> b;
        t[i] = a;
        c[i] = b;
    }
​
    for (int i = 0; i < m; i++) { //将每一个查询的时刻进行保存
        int d = 0;
        cin >> d;
        q[i] = d;
    }
​
    int res = 0; //记录共有多少个项计划可以满足
​
    for (int i = 0; i < m; i++) { //循环每个查询
        for (int j = 0; j < n; j++) { //查询每个时刻是否满足
            //进行区间判断：查询时间+等待时间
            if (q[i] + k <= t[j] && q[i] + k + c[j] - 1 >= t[j]) {
                res += 1;
            }
        }
        cout << res << endl;
        res = 0; //输出之后重置为0，第二次重新统计
    }
    return 0;
}

改进算法，使用查分数组求解，时间复杂度：O(n log n + m log m)

提交后满分100分

#include <bits/stdc++.h>
using namespace std;

const int N = 4e5+10;  // 定义常量N为400010，这是数组的最大长度

int main() {
    int n, m, k;
    int b[N];  // 定义数组b，用于实现差分数组

    cin >> n >> m >> k;  // 输入出行计划数目n，查询数目m，以及等待核酸检测结果的时间k

    int t, c;
    for (int i = 1; i <= n; i++) {  // 读入每一个出行计划
        cin >> t >> c;  // t是出行时间，c是需要的核酸检测证明的有效时长
        int left = t - c + 1;  // 计算出行计划对应的检测起始有效时间
        left = left > 0 ? left : 1;  // 如果计算出的时间小于1，则取1
        int right = t;  // 计算出行计划对应的检测结束有效时间
        b[left]++;  // 在差分数组中，left位置加1
        b[right + 1]--;  // 在差分数组中，right的下一个位置减1
    }

    for (int i = 1; i < N; i++) {  // 对差分数组进行前缀和处理，以计算每个时间点的实际有效出行计划数
        b[i] = b[i - 1] + b[i];
    }

    for (int i = 1; i <= m; i++) {  // 对每个查询进行处理
        int q;
        cin >> q;  // 输入查询的时间点
        cout << b[q + k] << endl;  // 输出该时间点进行核酸检测后，可以满足的出行计划数量
    }

    return 0;
}