10180. 「一本通 5.5 练习 1」烽火传递-单调队列

最新推荐文章于 2021-03-16 11:23:04 发布

lengxuenong

最新推荐文章于 2021-03-16 11:23:04 发布

阅读量301

点赞数

CC 4.0 BY-SA版权

分类专栏： dp

本文链接：https://blog.youkuaiyun.com/lengxuenong/article/details/104960992

dp 专栏收录该内容

26 篇文章

订阅专栏

本文分享了一次使用单调队列优化动态规划(DP)算法的经历。通过对比两种实现方式，探讨了代码规范的重要性，尤其是在函数返回值类型的选择上。文章详细展示了如何利用单调队列来减少DP状态转移的时间复杂度，提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一定要注意代码规范，函数没有返回值就VOID

写了个单调队列，本机各种修改，都是对的，交上去就TLE，RE.刷了两页提交记录后，重于发现原来函数想写返回值，就用了int，后来不打算返回值了，也没有修改。

dp[i]=ai+min(dp[j]) i-j<k;

评测总是不过，所以用两种方法写的

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=2e5+10;
struct node {
	int id;
	ll x;
} q[N];
ll a[N],sum[N],dp[N];
ll n,k,head=1,tail=0,ans=2e9;
void inq(int i) { //队首最小值，单调增
	
	while(head<=tail&&q[head].id+k<=i)head++;
	while(head<=tail&&q[tail].x>=dp[i-1])tail--;
	q[++tail].x=dp[i-1],q[tail].id=i;
	dp[i]=q[head].x+a[i];
	if(n-i<k)ans=min(ans,dp[i]);
}
int main() {
	//dp[i]=ai+min(dp[j)):j+k<=i;
	scanf("%lld%lld",&n,&k);
	for(int i=1; i<=n; i++) {
		scanf("%lld",&a[i]);
	}
	memset(dp,63,sizeof(dp));
	dp[0]=0;
	for(int i=1; i<=n; i++) {
		inq(i);
	}
	cout<<ans<<endl;
}

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
struct node {
	int id,x;
} ;
deque<node>q;
int a[N],dp[N];
int n,k,head=1,tail=1,ans=2e9;
void inq(int i) { //队首最小值，单调增	
	while(q.size()&&q.front().id<i-k)q.pop_front();	
	while(q.size()&&q.back().x>dp[i-1])q.pop_back();
	q.push_back({i-1,dp[i-1]});
	dp[i]=q.front().x+a[i];
}
int main() {
	//dp[i]=ai+min(dp[j)):j+k<=i;
	scanf("%d%d",&n,&k);
	for(int i=1; i<=n; i++) {
		scanf("%d",&a[i]);
	}
	for(int i=1; i<=n; i++) {
		inq(i);
	}
	inq(n+1);
	cout<<q.front().x<<endl;
	
}