ST表优化dp求助 悬关
  • 板块学术版
  • 楼主___A__
  • 当前回复0
  • 已保存回复0
  • 发布时间2023/11/11 22:52
  • 上次更新2023/11/11 22:54:25
查看原帖
ST表优化dp求助 悬关
275860
___A__楼主2023/11/11 22:52

CF1077F2 ST表是对的 只有dp错了

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N=5e3+5,inf=1e15;
int dp[N][N],n,m,k,a[N],ST[N][15];//dp[i][j]:前i个选j个 一定选第i个的答案
int query(int l,int r){
	int bit=__lg(r-l+1);
	return max(ST[l][bit],ST[r-(1<<bit)+1][bit]);
}
signed main(){
	cin>>n>>k>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	if(n/k>m){
		cout<<-1;
		return 0;
	}
	for(int i=0;i<=n;i++){
		for(int j=0;j<=m;j++){
			dp[i][j]=-inf;
		}
	}
	dp[0][0]=0;
	int ans=0;
	for(int j=1;j<=m;j++){
		for(int i=0;i<=n;i++){
			ST[i][0]=dp[i][j-1];
		}
	    for(int bit=1;bit<15;bit++){
	    	for(int i=0;i+(1<<bit)-1<=n;i++){
	    		ST[i][bit]=max(ST[i][bit-1],ST[i+(1<<(bit-1))][bit-1]);
			}
		}
		for(int i=1;i<=n;i++){
			dp[i][j]=a[i]+query(max(0ll,i-k),i-1);
			ans=max(ans,dp[i][j]);
		}
	}
	cout<<ans;
}
2023/11/11 22:52
加载中...