【双指针优化DP】CF985E

lamentropetion

已于 2023-07-29 01:10:19 修改

阅读量183

点赞数

分类专栏：动态规划 Tricks 文章标签：算法

于 2023-07-28 10:57:08 首次发布

本文链接：https://blog.csdn.net/weixin_62528401/article/details/131975890

版权

动态规划同时被 2 个专栏收录

144 篇文章 2 订阅

订阅专栏

Tricks

61 篇文章 0 订阅

订阅专栏

文章介绍了如何解决Codeforces上的Problem-985E问题，通过排序后利用双指针进行分段动态规划（DP）的方法，以优化从n^2到线性的复杂度。关键在于识别问题的单调性，避免不必要的回溯，以确保在极差小于给定值D的情况下，找到相距K的位置。

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

Problem - 985E - Codeforces

题意：

思路：

首先肯定需要排个序，然后分段DP

但是平常写的分段DP都是n^2的，这里的可以发现单调性，因此考虑双指针转移

这里两个指针都代表两个段的右端点

这里双指针的合理性很明显

想象指针为了满足位置相距K，先j=max(j,i+K)了

然后还得满足极差<=D，因此需要移动几个距离

合理性重点在于当另一个指针i往前移动一格时，j有没有必要回溯

答案是没有必要，因为回溯会导致极差更小，而相距距离也更小了

Code：

#include <bits/stdc++.h>

//#define int long long

using namespace std;

const int mxn=5e5+10;

int N,K,D;
int a[mxn],dp[mxn];

void solve(){
	cin>>N>>K>>D;
	for(int i=1;i<=N;i++) cin>>a[i];
	sort(a+1,a+1+N);
	int j=1;
	dp[0]=1;
	for(int i=0;i<=N;i++){
		if(dp[i]){
			j=max(j,i+K);
			while(j<=N&&a[j]-a[i+1]<=D) dp[j]=1,j++;
		}
	}
	if(dp[N]==1) cout<<"YES"<<'\n';
	else cout<<"NO"<<'\n';
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	int __=1;//cin>>__;
	while(__--)solve();return 0;
}