深圳 微网站百度网页版登录入口官网
题目
假定有一个无限长的数轴,数轴上每个坐标上的数都是 0。
现在,我们首先进行 n 次操作,每次操作将某一位置 x 上的数加 c。
接下来,进行 m 次询问,每个询问包含两个整数 l 和 r,你需要求出在区间 [l,r] 之间的所有数的和。
输入格式
第一行包含两个整数 n 和 m。
接下来 n 行,每行包含两个整数 x 和 c。
再接下来 m 行,每行包含两个整数 l 和 r。
输出格式
共 m 行,每行输出一个询问中所求的区间内数字和。
数据范围
−109≤x≤109
1≤n,m≤105
−109≤l≤r≤109
−10000≤c≤10000
输入样例:
3 3
1 2
3 6
7 5
1 3
4 6
7 8
输出样例:
8
0
5
来源:acwing算法基础 802. 区间和
思路(注意事项)
N初始化为3e5 + 10 是因为操作最大1e5次,查询 2 * 1e5次。
纯代码
#include<bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10;
int a[N], alls[N];
vector<int> all;
vector<pair<int,int>> add, query;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;
}
int main()
{int n, m;cin >> n >> m;while (n -- ){int x, c;scanf ("%d %d", &x, &c);all.push_back(x);add.push_back({x, c});}while (m --){int l, r;scanf ("%d %d", &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 i : add){int t = find (i.first);a[t] += i.second;}for (int i = 1; i <= all.size(); i ++) // 前缀和 alls[i] = alls[i - 1] + a[i];for (auto i : query){int a = find (i.first), b = find (i.second);cout << alls[b] - alls[a - 1] << endl;}return 0;
}
题解(带注释)
#include<bits/stdc++.h>
using namespace std;const int N = 3e5 + 10; // 定义常量N,表示最大数据范围
int a[N], alls[N]; // a数组存储离散化后的值,alls数组存储前缀和
vector<int> all; // all存储所有需要离散化的值
vector<pair<int, int>> add, query; // add存储操作,query存储查询// 二分查找函数:在离散化后的数组all中查找x的位置,并返回其在alls数组中的下标(从1开始)
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; // 如果中间值大于等于x,缩小右边界else l = mid + 1; // 否则缩小左边界}return r + 1; // 返回下标(从1开始)
}int main()
{int n, m;cin >> n >> m; // 输入n和m,表示操作次数和查询次数// 处理n个操作while (n--){int x, c;scanf("%d%d", &x, &c); // 输入x和call.push_back(x); // 将x存入all数组add.push_back({x, c}); // 将操作存入add数组}// 处理m个查询while (m--){int l, r;scanf("%d%d", &l, &r); // 输入l和rquery.push_back({l, r}); // 将查询存入query数组all.push_back(l); // 将l存入all数组all.push_back(r); // 将r存入all数组}// 离散化处理sort(all.begin(), all.end()); // 对all数组排序all.erase(unique(all.begin(), all.end()), all.end()); // 去重// 处理操作for (auto i : add){int t = find(i.first); // 找到x离散化后的位置a[t] += i.second; // 在a数组中累加c}// 计算前缀和for (int i = 1; i <= all.size(); i++) // 遍历alls数组alls[i] = alls[i - 1] + a[i]; // 计算前缀和// 处理查询for (auto i : query){int a = find(i.first), b = find(i.second); // 找到l和r离散化后的位置cout << alls[b] - alls[a - 1] << endl; // 输出区间和}return 0; // 程序结束
}