借教室

链接

信息提取:首先给出n天(也就是数据长度),和m个订单(也就是操作次数)

接下来有一行是n个数据 d1 d2 d3 ... dn

在接下来m行,每行有三个数据,分别为教室数量 val, 初始时间l(左区间),归还时间r(右区间)

题目的要求就是让我们判断dl~dr的数据是否大于每每行val,如果某一行时小于val,让我们输出-1和订单编号 否则输出0

这题的思路很简单,就是直接让给出的n个数据在执行订单时减去val,如果出现负数,直接返回

但是,由于数据量很大,直接暴力做减法会超时

因此,我们可以用线段树来解决这个问题

线段树的优势就是对某一数据进行操控(增大或减小)时,能快速得到某区间的最大或最小值

而操控数据就是减去value,只要我们发现最小值小于0,那么直接退出

这个速度比暴力减法节省好多时间

#include<iostream>
#include<vector>
#include<algorithm>
#include<climits>
#define ll long long
using namespace std;
class node {
private:
	vector<ll>tree, lazy;
	int n;
	void build(vector<ll>& data, int left, int right, int root) {
		if (left == right) {
			tree[root] = data[left];
			return;
		}
		ll mid = left + (right - left) / 2;
		build(data, left, mid, 2 * root);
		build(data, mid + 1, right, 2 * root + 1);
		tree[root] = min(tree[2 * root], tree[2 * root + 1]);
	}
	void push_down(int root) {
		if (lazy[root]) {
			int left = 2 * root;
			int right = 2 * root + 1;
			tree[left] += lazy[root];
			tree[right] += lazy[root];
			
			lazy[left] += lazy[root];
			lazy[right] += lazy[root];
			lazy[root] = 0;
		}
	}
	void update(int root, int left, int right, int ql, int qr, ll value) {
		if (qr<left || ql>right) {
			return;
		}
		if (qr >= right && ql <= left) {
			tree[root] += value;
			lazy[root] += value;
			return;
		}
		push_down(root);
		ll mid = left + (right - left) / 2;
		update(2 * root, left, mid, ql, qr, value);
		update(2 * root + 1, mid + 1, right, ql, qr, value);
		tree[root] = min(tree[2 * root], tree[2 * root + 1]);
	}
	ll query_min(int root, int left, int right, int ql, int qr) {
		if (qr<left || ql>right) {
			return LLONG_MAX;
		}
		if (ql <= left && qr >= right) {
			return tree[root];
		}
		push_down(root);
		ll mid = left + (right - left) / 2;
		ll leftMin = query_min(2 * root, left, mid, ql, qr);
		ll rightMin = query_min(2 * root + 1, mid + 1, right, ql, qr);
		return min(leftMin, rightMin);
	}
public:
	node(vector<ll>&data) {
		 n = data.size()-1;
		tree.resize(4 * (n+1));
		lazy.resize(4 * (n+1), 0);
		build(data, 1, n , 1);
	}
	bool range_subtract(int left, int right, ll value) {
		ll min_num = query_min(1, 1, n , left, right);
		if (min_num < value) {
			return false;
		}
		else {
			update(1, 1, n , left, right, -value);
			return true;
		}
	}
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, m;
	cin >> n >> m;
	vector<ll>data(n+1);

	
	for (int i = 1;i <= n;i++) {
		cin >> data[i];
	}
	node Tree(data);
	for(int i=0;i<m;i++) {
		ll value;
		int l, r;
		cin >> value >> l >> r;
		if (!Tree.range_subtract(l, r, value)) {
			cout << -1 << '\n';
			cout << i + 1 << '\n';
			return 0;
		}
	}
	cout << 0 << '\n';
	return 0;
}

时间复杂度O:(n + m log n) 直接做减法是m x n

空间复杂度:O(n)

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务