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