CF-1011-D. Serval and Kaitenzushi Buffet

思路讲解

K其实就是吃掉一个寿司所需要花费的时间。
你可以不停拿，但是最后要吃完，拿的时候不能吃。

这个其实是个贪心，这个把题目完全读完就可以得出。

https://www.luogu.com.cn/article/flnzpg4m 这个题解写的很清楚。

总的来说，其实就是这么选一定是合法的，但不一定最优。

而且可以证明所有的O是不能再往后移了，只能往前移。当然，往前移一定是合法的。

具体而言是这样的。

ll idx=0;
ROF(i,N,1){
	++idx;
	if(idx%(K+1)==0){
		spec[i]=true;
	}
}
priority_queue<ll> pq;
ll ans=0;
FOR(i,1,N){
	pq.push(D[i]);
	if(spec[i]){
		ans+=pq.top();
		pq.pop(); // 记得要弹出，不用清空，往前移比较多也是可以接受的。
	}
}

AC代码

https://codeforces.com/contest/2085/submission/319924135

源代码

// Problem: D. Serval and Kaitenzushi Buffet
// Contest: Codeforces - Codeforces Round 1011 (Div. 2)
// URL: https://codeforces.com/contest/2085/problem/D
// Memory Limit: 256 MB
// Time Limit: 2000 ms
// by znzryb
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
#define all(vec) vec.begin(),vec.end()
#define CLR(i,a) memset(i,a,sizeof(i))
#define fi first
#define se second
#define pb push_back
#define SZ(a) ((int) a.size())
#define FOR(i, a, b) for (int i = (a); i <= (b); ++i)
#define ROF(i, a, b) for (int i = (a); i >= (b); --i)
#define getFro(vec) (vec.empty()?0:vec.front())
#define getBac(vec) (vec.empty()?0:vec.front())
#define debug(var) cerr << #var <<":"<<var<<"\n";
#define DEBUG(variable) \
    do { \
        std::cerr << #variable << ":"; \
        for (const auto& elem : variable) { \
            std::cerr << elem << " "; \
        } \
        std::cerr << "\n"; \
    } while (0)
#define uniVec(var) \
    do { \
        sort(var.begin(),var.end());\
        var.resize(unique(var.begin(),var.end())-var.begin());\
    } while (0)
#define debugN(var,N) \
	do{ \
		std::cerr<<#var<<":"; \
		FOR(i,1,N){ \
			std::cerr<<var[i]<<" "; \
		} \
		std::cerr<<"\n"; \
	}while(0)
#define debugMap(variable) \
    do { \
        std::cerr << #variable << ":\n"; \
        for (const auto& pair : variable) { \
            std::cerr << "  " << pair.first << " => " << pair.second << "\n"; \
        } \
        std::cerr << std::endl; \
    } while (0)
#define lson(var) (var<<1)
#define rson(var) ((var<<1)+1)

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef __int128 i128;
typedef pair<ll,ll> pll;
typedef array<ll,3> arr;
typedef double DB;
typedef pair<DB,DB> pdd;
typedef pair<ll,bool> plb;
constexpr ll MAXN=static_cast<ll>(1e6)+10,INF=static_cast<ll>(1e18)+3;
constexpr ll mod=static_cast<ll>(1e9)+7;
constexpr double eps=1e-8;

ll N,M,K,T,D[MAXN];
bool spec[MAXN];
/*
K其实就是吃掉一个寿司所需要花费的时间。
你可以不停拿，但是最后要吃完，拿的时候不能吃。

*/
inline void solve(){
	cin>>N>>K;
	FOR(i,0,N+5){
		D[i]=0;
		spec[i]=false;
	}
	FOR(i,1,N){
		cin>>D[i];
	}
	ll idx=0;
	ROF(i,N,1){
		++idx;
		if(idx%(K+1)==0){
			spec[i]=true;
		}
	}
	priority_queue<ll> pq;
	ll ans=0;
	FOR(i,1,N){
		pq.push(D[i]);
		if(spec[i]){
			ans+=pq.top();
			pq.pop();
		}
	}
#ifdef LOCAL
    debugN(spec,N);
#endif
	cout<<ans<<"\n";
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	cin>>T;
	while(T--){
		solve();
	}
	// solve();
	return 0;
}
/*
AC https://codeforces.com/contest/2085/submission/319924135
*/

心路历程（WA，TLE，MLE……）