例题:区间和:
假定有一个无限长的数轴,数轴上每个坐标上的数都是0。现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c。接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r]之间的所有数的和。
输入格式
第一行包含两个整数 n 和 m。接下来 n 行,每行包含两个整数 x 和 c。再接下来 m 行,每行包含两个整数 l 和 r。
输出格式
共 m行,每行输出一个询问中所求的区间内数字和。
数据范围
−1000000000≤x≤1000000000,
1≤n,m≤100000,
−1000000000≤l≤r≤1000000000,
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
首先:
离散化(Discretization),把无限空间中有限的个体映射到有限的空间中去,以此提高算法的时空效率。通俗的说,离散化是在不改变数据相对大小的条件下,对数据进行相应的缩小。
例如:5 5000 2000 3000 —>1 4 2 3 和 {100, 200},{20, 50000},{1, 400}—>{3,4},{2,6},{1,5}。这样的过程。
其中的unique为除重函数,erase是将区间中的数删除。
假设有a[100];那么a.erase(a.begin()+5,a.end()-2);即删除第6~97个元素的值,下面附上erase的解释代码:
vector<string> e = {"a","b","c","d","e","f","g"};
e.erase("c"); //删除字符串“c”
auto it = e.end()-1; //.end()指向末尾的后一个元素,因此需要-1,指向末尾元素.
e.erase(it); //删除末尾元素“e”.
auto x1 = e.begin()+1;
auto x2 = e.end()-2;
e.erase(x1,x2); //删除x1到x2之间的元素.
注:a.begin(),a.end()来自函数库<vector>.erase 来自函数库<algorithm>.
在这段代码中:
for (auto item : add)
{
int x = find(item.first);
a[x] += item.second;
}
其中:item为一个迭代器表示从add这个数组的第一个元素循环到add的最后一个元素结束循环,auto为检测add数组数据类型,并且自动转化为相应的类型。其中的item.first表示前面输入的x和c中的x,调用find函数来找到x的位置并且对a[x]加上item.second即c的值,这样就能做到第一步操作:将某一位置 x 上的数加 c。
同理下面的:query也是一样的道理,来进行区间求和。
此题中a数组得元素初始全部为零,若不为零则应在最前面对数组a进行赋值
下面放代码:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> PII;//pair<int,int>类似于结构体的简写版,typedef定义了一个新的类型PII来调用两个<int>类型的数据。
const int N = 300010;//坐标x的数量上限为1e5,两个坐标l,r的数量上限也为1e5,所以加起来为3*le5;
int n, m;
int a[N], s[N];
vector<int> all;//定义整数类型的数组all;
vector<PII> add, query;//定义俩个PII型的数据类型来进行插入和区间求和
//二分查找位置
int find(int x)
{
int l = 0, r = all.size() - 1;
while (l < r)
{
int mid = l + r >> 1;
if (all[mid] >= x) r = mid;
else l = mid + 1;
}
return r + 1;//由于使用后缀和时是从1开始循环,故要写成j+1;
}
int main()
{
cin >> n >> m;
for (int i = 0; i < n; i ++ )
{
int x, c;
cin >> x >> c;
add.push_back({x, c});
all.push_back(x);
}
for (int i = 0; i < m; i ++ )
{
int l, r;
cin >> l >> r;
query.push_back({l, r});
all.push_back(l);
all.push_back(r);
}
// 去重+排序
sort(all.begin(), all.end());
all.erase(unique(all.begin(), all.end()), all.end());
// 处理插入
for (auto item : add)
{
int x = find(item.first);//调用二分函数查找x在all数组中的位置;
a[x] += item.second;//为数组中的第x个数加上c;
}
// 预处理前缀和。在使用时就可以直接调用(其实就是对s[i]数组赋值)来代表前i个数的和
for (int i = 1; i <= all.size(); i ++ ) s[i] = s[i - 1] + a[i];
// 处理询问
for (auto item : query)
{
int l = find(item.first), r = find(item.second);//分别找到两个区间在all数组中的对应位置;
cout << s[r] - s[l - 1] << endl;//此时在前面已经对s数组赋值,在此可以直接使用来求得对应区间的和;
}
return 0;
}
此代码来自作者:yxc
来源:AcWing
本文大部分思想来自本人,代码与题目来自ACwing网站,yxc大佬之手!!!