【DP】【线段树优化】Day3 - A

最新推荐文章于 2024-07-30 13:57:15 发布

ssllth

最新推荐文章于 2024-07-30 13:57:15 发布

阅读量91

点赞数

分类专栏： DP 线段树文章标签：动态规划 01背包线段树

本文链接：https://blog.csdn.net/LTH060226/article/details/119611084

版权

DP 同时被 2 个专栏收录

103 篇文章 2 订阅

订阅专栏

线段树

11 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种线段树在解决01背包问题中的应用。题目中给出了n个物品和m个人，每个人可以在特定范围内删除物品，删除的代价不同。通过线段树动态更新删除代价，并利用01背包的思路求解最大总价值。代码中展示了如何使用线段树实现这一过程。

摘要由CSDN通过智能技术生成

小目录

链接

题目描述

给出n个物品和m个人，定义总价值为所有物品的价值之和，每个人可以在区间[l,r]的范围内删掉若干个数，删掉一个数的代价为ki，最多可以承受c的代价，问最大总价值为多少

样例输入

5 5 10
10 -2 -5 7 -10
1 1 5
2 4 10
4 4 12
3 4 10
1 5 15

样例输出

思路

发现就是一个01背包，但是赋值代价的时候有点麻烦，因为n^2会爆，那我就用线段树去跑
$f_{j} = f_{j-w_i}+v_i$

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long

using namespace std;

int we[100005], tr[800005], n, m, k, l, r, w;
ll b[100005], f[5005], ans = -1e9, sum;

void change(int x, int l, int r, int L, int R, int w)
{
	if(l == r){
		tr[x] = min(tr[x], w);
		return;
	}
	int mid = (l + r) >> 1;
	if(L <= mid) change(x * 2, l, mid, L, R, w); 
	if(mid < R) change(x * 2 + 1, mid + 1, r, L, R, w);
}//修改单点的代价

int ask(int x, int l, int r, int L, int R)
{
	if(L <= l && r <= R) return tr[x];
	int mid = (l + r) >> 1;
	if(L <= mid) return ask(x * 2, l, mid, L, R);
	if(mid < R) return ask(x * 2 + 1, mid + 1, r, L, R);
}//查询单点的代价

int main()
{
	memset(tr, 0x3f, sizeof(tr));
	scanf("%d%d%d", &n, &m ,&k);	
	for(int i = 1; i <= n; ++i)
		scanf("%lld", &b[i]), sum += b[i];
	for(int i = 0; i <= 500; ++i)
		f[i] = sum;
	for(int i = 1; i <= m; ++i)
	{
		scanf("%d%d%d", &l, &r, &w);
		change(1, 1, n, l, r, w);
	}
	for(int i = 1; i <= n; ++i) 
		we[i] = ask(1, 1, n, i, i);
	for(int i = 1; i <= n; ++i)
		for(int j = k; j >= 0; --j)
			if(j - we[i] >= 0) 
				f[j] = max(f[j], f[j - we[i]] - b[i]);//01背包
	for(int i = 0; i <= k; ++i)
		ans = max(ans, f[i]);
	printf("%lld", ans);
	return 0;
}