結果

問題 No.3588 Already Ready
コンテスト
ユーザー askr58
提出日時 2026-07-10 22:53:16
言語 C++23
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
AC  
実行時間 70 ms / 3,000 ms
コード長 1,544 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,482 ms
コンパイル使用メモリ 210,280 KB
実行使用メモリ 12,940 KB
最終ジャッジ日時 2026-07-10 22:53:23
合計ジャッジ時間 4,861 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 69
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <queue>
#include <vector>
#include <atcoder/lazysegtree>
using namespace std;
using ll=long long;
using P=pair<int,int>;
P op(P x,P y){
	return min(x,y);
}
int inf=1e9;
P e(){
	return make_pair(inf,inf);
}
P mapping(int a,P x){
	x.first+=a;
	return x;
}
int composition(int x,int y){
	return x+y;
}
int id(){
	return 0;
}
int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int n,k,m;
	cin>>n>>k>>m;
	m--;
	vector<int> a(n);
	for(int i=0;i<n;i++)cin>>a[i];
	vector<int> ans;
	ans.push_back(m);
	for(int i=0;i<n;i++){
		if(i==m)a[i]-=2;
		else a[i]--;
		if(i==m&&a[i]<k){
			cout<<-1<<endl;
			return 0;
		}
	}
	ll sum=0;
	for(int i=0;i<n;i++)sum+=a[i];
	if(sum%(n+1)!=0){
		cout<<-1<<endl;
		return 0;
	}
	sum/=n+1;
	for(int i=0;i<n;i++){
		if(a[i]<sum){
			cout<<-1<<endl;
			return 0;
		}
	}
	atcoder::lazy_segtree<P,op,e,int,mapping,composition,id> seg(n);
	priority_queue<P> pq;
	for(int i=0;i<n;i++){
		if(a[i]<=k+1){
			if(a[i]>sum)pq.push(make_pair(i,a[i]-sum));
		}else{
			seg.set(i,make_pair(a[i]-k-1,i));
		}
	}
	for(int t=0;t<sum;t++){
		while(seg.all_prod().first<1e8){
			auto[x,y]=seg.all_prod();
			if(x==0){
				if(a[y]>sum)pq.push(make_pair(y,a[y]-sum));
				seg.set(y,e());
			}else break;
		}
		if(pq.empty()){
			cout<<-1<<endl;
			return 0;
		}
		auto[x,y]=pq.top();
		pq.pop();
		seg.apply(0,n,-1);
		ans.push_back(x);
		if(y>1)pq.push(make_pair(x,y-1));
	}
	cout<<ans.size()<<endl;
	for(int i=ans.size()-1;i>=0;i--){
		cout<<ans[i]+1;
		if(i>0)cout<<" ";
		else cout<<endl;
	}
}
0