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;
	
}

 

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