题目

  • 假定有一个无限长的数轴,数轴上每个坐标上的数都是$ 0$。
  • 现在,我们首先进行 $n $次操作,每次操作将某一位置 $x$ 上的数加 $c$。
  • 接下来,进行 $m$ 次询问,每个询问包含两个整数 $l$ 和 $r$,你需要求出在区间 $[l,r]$ 之间的所有数的和。

输入格式

  • 第一行包含两个整数 $n $和 $m$。
  • 接下来 $n$ 行,每行包含两个整数 $x $和 $c$。
  • 再接下来$ m$ 行,每行包含两个整数 $l$ 和 $r$。

输出格式

  • 共 $m$ 行,每行输出一个询问中所求的区间内数字和。

本文题目:区间和

代码详解

$\huge \color{red}{离散化求区间和代码详解}$

  • $find函数是实现离散化映射的关键$
int find(int x)
{
    int l = 0, r = alls.size() - 1;
    while(l < r)
    {
        int mid = l + r >> 1;
        if(alls[mid] >= x)  r = mid;    // 找到的是第一个 >=x 的数
        else    l = mid + 1;
    }
    return r +  1;  // 根据题意返回值,我们的映射从 1 开始
}
  • $对 add 操作进行收集$
    int x, c;
    for (int i = 0; i < n; i ++ ) {
        cin >> x >> c;
        alls.push_back(x);
        add.push_back({x, c});
    }
  • $对 query 操作进行收集 $
    int l, r;
    for (int i = 0; i < m; i ++ ) {
        cin >> l >> r;
        query.push_back({l, r});
        
        alls.push_back(l);
        alls.push_back(r);
    }
  • $排序 + 去重$
    sort(alls.begin(), alls.end());
    alls.erase(unique(alls.begin(), alls.end()), alls.end());

如果没有$unique$函数该怎么办??

我们势必要自己去实现------$\color{red}{双指针思想}$


// 定义去重函数 unique

vector<int>::iterator unique(vector<int>& v) {
    int i, j;
    for (i = 0, j = 0; i < v.size(); i ++ ) {
        if(!i || v[i] != v[i - 1]) 
            v[j ++ ] = v[i];
    }
    // 从 0 ~ j-1 已去重
    return v.begin() + j;   // 返回已去重的右端点的下一个坐标
}

  • $处理插入$
    for (auto item : add) {
        int x = find(item.first);
        a[x] += item.second;
    }
  • $预处理前缀和$

$ for (int\space i\space =\space 1;\space i\space <=\space alls.size();\space i\space ++\space )\space\space\space s[i]\space =\space s[i - 1]\space +\space a[i];$

  • $处理查询$
    for (auto item : query) {
        // 实际作用的是 alls 映射后的单元
        int r = find(item.second), l = find(item.first);
        cout << s[r] - s[l - 1] << endl;
    }

$\huge \color{blue}{完整代码:}$

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

using namespace std;

typedef pair<int, int> PA;

const int N = 3e5 + 10;
int a[N], s[N];     // s[] 为 a[] 的前缀和

vector<int> alls;
vector<PA> add, query;

int find(int x)
{
    int l = 0, r = alls.size() - 1;
    while(l < r)
    {
        int mid = l + r >> 1;
        if(alls[mid] >= x)  r = mid;    // 找到的是第一个 >=x 的数
        else    l = mid + 1;
    }
    return r +  1;  // 根据题意返回值,我们的映射从 1 开始
}

int main()
{
    int n, m;
    cin >> n >> m;
    
    // 对 add 操作进行收集
    int x, c;
    for (int i = 0; i < n; i ++ ) {
        cin >> x >> c;
        alls.push_back(x);
        add.push_back({x, c});
    }

    // 对 query 操作进行收集    
    int l, r;
    for (int i = 0; i < m; i ++ ) {
        cin >> l >> r;
        query.push_back({l, r});
        
        alls.push_back(l);
        alls.push_back(r);
    }
    
    // 排序 + 去重
    sort(alls.begin(), alls.end());
    alls.erase(unique(alls.begin(), alls.end()), alls.end());
    
    // 处理插入
    for (auto item : add) {
        int x = find(item.first);
        a[x] += item.second;
    }
    
    // 预处理前缀和
    for (int i = 1; i <= alls.size(); i ++ )     s[i] = s[i - 1] + a[i];
    
    // 处理查询
    for (auto item : query) {
        // 实际作用的是 alls 映射后的单元
        int r = find(item.second), l = find(item.first);
        cout << s[r] - s[l - 1] << endl;
    }
    
    return 0;
}

最后修改:2023 年 01 月 08 日 12 : 38 PM
如果我的文章对你有用,请随意赞赏