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

一定要注意代码规范,函数没有返回值就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;
	
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章