【离散化】区间和

题目链接

活动 - AcWing

• 这题给出数据范围
• 虽然x,l,r的区间长度为2e9
• 但是实际会占用数组的容量
• 仅仅有n=1e5,m=1e5
• 共n+2*m=3e5 个

 解释

• 开辟三个数组容器 alls和add和query
• alls用来存储x,l,r
• add用来存储数对{x,c}方便后续给离散化后的数列进行加C操作
• query用来存储每次询问{l,r},并将l和r也放入alls中
• 如果不传入l,r，有可能没有对应的x=l或x=r，这样就询问不到这个区间段的内容了
• 进行排序与去重操作
• 如果有l=x或r=x，则可能出现重复的元素
• 使用一个手写unique将所有重复元素放到数组最后，并返回一个首地址进行清除
• 去重的手写原理
• 使用双指针i和j
• 如果a[i]是第一个元素或者a[i]!=a[i-1]
• 就执行a[i]=a[j++]
• 这样就能将不同的元素全部放到(a.begin(),a.begin()+j-1)这个区间段了
• 最后返回需要清除的元素首地址a.begin()+j
• 进行查找与加C操作
• 查找的手写方法是二分法，在alls里定位a[mid]=x
• 查找函数里返回r+1是让离散化后的点从1开始，方便操作
• 然后再在a[find]加上C（此时已经将x离散化到a数组中）
• 数组all到a的一种哈希映射关系，映射函数为下标
• 预处理前缀和
• 将a[i]中的元素前缀和放入s[i]中，方便询问{l,r}直接得到区间内容
• 进行询问操作
• 先获取l,r离散化后kl和kr
• 然后计算s[kl]-s[kr-1]的值 即所求区间内容

代码段

typedef pair<int, int>PII;
vector<int>alls;
vector<PII>add, query;
int n,m;
const int N = 300010;
int a[N], s[N];
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;
        else
            l = mid + 1;
    }
    return r + 1;
}
vector<int>::iterator unique(vector<int>& a)
{
    int j = 0;
    for (int i = 0; i < a.size(); i++)
        if (!i || a[i] != a[i - 1])
            a[j++] = a[i];
    return a.begin() + j;
}
int main()
{
    cin >> n >> m;
    for (int i = 0; i < n; i++)
    {
        int x, c;
        cin >> x >> c;
        add.push_back({ x,c });
        alls.push_back(x);
    }
    for (int i = 0; i < m; i++)
    {
        int l, r;
        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), 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)
    {
        int l = find(item.first), r = find(item.second);
        cout << s[r] - s[l - 1] << endl;
    }
    return 0;
}