ABC426G - Range Knapsack Query

G - Range Knapsack Queryhttps://atcoder.jp/contests/abc426/tasks/abc426_g

题意 :

给你Q个查询,每次查询要求你在l~r中进行一次01背包

题解 :

直接计算时O(qn) 的,肯定过不了,我们可以考虑cdq分治之类的离线处理

构建一个分治函数solve(l,r),我们可以将查询分为以下三类 :

**1.**询问区间在分治区间中点左边的

**2.**询问区间在分治区间中点右边的

**3.**询问区间横跨分治区间中点的

对于前两类,我们可以分治处理,所以重点在第三类

因为询问区间包含了mid,所以我们可以分别计算l~mid 的和mid+1~r的,对其分别采取背包

cpp 复制代码
for(int i=mid;i>=l;i--)
{
	for(int j=0;j<=500;j++)
	{
  		dp[0][i][j]=dp[0][i+1][j];
		if(j>=w[i]){
			dp[0][i][j]=max(dp[0][i][j],dp[0][i+1][j-w[i]]+v[i]);
		}
	}
}
for(int i=mid+1;i<=r;i++)
{
	for(int j=0;j<=500;j++)
	{
		dp[1][i][j]=dp[1][i-1][j];
		if(j>=w[i]){
			dp[1][i][j]=max(dp[1][i][j],dp[1][i-1][j-w[i]]+v[i]);
		}
	}
}

然后再考虑将其合并

cpp 复制代码
for(int i=1;i<=m;i++){
	if(d[i].l<=mid&&d[i].r>mid){
		for(int j=0;j<=d[i].c;j++){
			ans[d[i].id]=max(ans[d[i].id],dp[d[i].l][j]+dp[d[i].r][d[i].c-j]);
		}
	}
}

但由于一次合并的时间复杂度是O(m) 的,所以总时间复杂度依旧是O(mn),所以我们需要优化

因为所有有效合并只有m次,所以我们可以设p,q表示本次合并包含的查询区间

cpp 复制代码
for(int i=p;i<=q;i++){
	if(d[i].l<=mid&&d[i].r>mid){
		for(int j=0;j<=d[i].c;j++){
			ans[d[i].id]=max(ans[d[i].id],dp[0][d[i].l][j]+dp[1][d[i].r][d[i].c-j]);
		}
	}
}

对于分治处理,我们需要将符合第一类和第二类的查询分别放入一个区间中

cpp 复制代码
int cnt=p-1;
for(int i=p;i<=q;i++){
	if(d[i].r<=mid){
		tmp[++cnt]=d[i];
	}
}
int cnt2=cnt;
for(int i=p;i<=q;i++){
	if(d[i].l>mid){
		tmp[++cnt2]=d[i];
	}
}
for(int i=p;i<=q;i++){
    d[i]=tmp[i];
}

这样就完成了solve函数

代码 :

cpp 复制代码
#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,m,w[21000],v[21000],dp[2][21000][510],ans[210000];
struct mmm{
	int l,r,c,id;
}d[210000],tmp[210000];
void solve(int l,int r,int p,int q){
	if(p>q) return;
	if(l==r){
		for(int i=p;i<=q;i++){
			ans[d[i].id]=(w[d[i].l]>d[i].c)?0:v[d[i].l];
		}
		return;
	}
	int mid=(l+r)>>1;
	for(int i=l;i<=r;i++){
		for(int j=0;j<=500;j++){
			dp[0][i][j]=dp[1][i][j]=0;
		}
	}
	for(int i=mid;i>=l;i--)
	{
		for(int j=0;j<=500;j++)
		{
			dp[0][i][j]=dp[0][i+1][j];
			if(j>=w[i]){
				dp[0][i][j]=max(dp[0][i][j],dp[0][i+1][j-w[i]]+v[i]);
			}
		}
	}
	for(int i=mid+1;i<=r;i++)
	{
		for(int j=0;j<=500;j++)
		{
			dp[1][i][j]=dp[1][i-1][j];
			if(j>=w[i]){
				dp[1][i][j]=max(dp[1][i][j],dp[1][i-1][j-w[i]]+v[i]);
			}
		}
	}
	for(int i=p;i<=q;i++){
		if(d[i].l<=mid&&d[i].r>mid){
			for(int j=0;j<=d[i].c;j++){
				ans[d[i].id]=max(ans[d[i].id],dp[0][d[i].l][j]+dp[1][d[i].r][d[i].c-j]);
			}
		}
	}
	int cnt=p-1;
	for(int i=p;i<=q;i++){
		if(d[i].r<=mid){
			tmp[++cnt]=d[i];
		}
	}
	int cnt2=cnt;
	for(int i=p;i<=q;i++){
		if(d[i].l>mid){
			tmp[++cnt2]=d[i];
		}
	}
	for(int i=p;i<=q;i++){
		d[i]=tmp[i];
	}
	solve(l,mid,p,cnt);
	solve(mid+1,r,cnt+1,cnt2);
}
signed main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld%lld",&w[i],&v[i]);
	}
	scanf("%lld",&m);
	for(int i=1;i<=m;i++){
		scanf("%lld%lld%lld",&d[i].l,&d[i].r,&d[i].c);
		d[i].id=i;
	}
	for(int i=1;i<=n;i++){
		if(d[i].l==d[i].r){
			ans[d[i].id]=(w[d[i].l]>d[i].c)?0:v[d[i].l];
		}
	}
	solve(1,n,1,m);
	for(int i=1;i<=m;i++){
		printf("%lld\n",ans[i]);
	}
	return 0;
}
相关推荐
爱凤的小光9 小时前
图漾新版看图软件操作手册(待完善版)
c++
CoovallyAIHub9 小时前
不看异常,怎么学会识别异常?用“异常”指导异常检测!——NAGL方法解析(附代码地址)
深度学习·算法·计算机视觉
共享家95279 小时前
数据结构-并查集
数据结构·c++·算法
AA陈超10 小时前
虚幻引擎5 GAS开发俯视角RPG游戏 P06-19 打开属性菜单
c++·游戏·ue5·游戏引擎·虚幻
IT古董10 小时前
【第五章:计算机视觉-项目实战之推荐/广告系统】2.粗排算法-(2)理解粗排模型之离线部分:双塔模型结构精讲及实现
人工智能·算法·计算机视觉
茉莉玫瑰花茶10 小时前
贪心 - 后篇
算法
m0_7482336410 小时前
【C++篇】C++11入门:踏入C++新世界的大门
java·c++·算法
lxmyzzs10 小时前
【图像算法 - 31】基于深度学习的太阳能板缺陷检测系统:YOLOv12 + UI界面 + 数据集实现
人工智能·深度学习·算法·yolo·缺陷检测
lxmyzzs10 小时前
【图像算法 - 32】基于深度学习的风力发电设备缺陷检测系统:YOLOv12 + UI界面 + 数据集实现
深度学习·算法·yolo·计算机视觉
m0_7482336410 小时前
jank实现C++无缝互操作的技术探索
开发语言·c++