比赛代码 - COGS

比赛	NOIP2025模拟赛1	评测结果	AAAAAAAAAAAAAAAAAAAA
题目名称	Ntarsis Set	最终得分	100
用户昵称	梦那边的美好ME	运行时间	0.279 s
代码语言	C++	内存使用	4.19 MiB
提交时间	2025-11-24 10:30:51

显示代码纯文本

#include<bits/stdc++.h>
using namespace std;
#define ll long long

ll n,k;
ll a[200010];
ll ans,cnt;

int main(){
    freopen("ntarsis_set.in","r",stdin);
    freopen("ntarsis_set.out","w",stdout);
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n>>k;
	for(int i=1;i<=n;i++) cin>>a[i];
	if(a[1]!=1){
        cout<<1<<'\n';
		return 0;
	}
	cnt=1;ans=1;
	for(int i=1;i<=k;i++){
		while(cnt<=n&&a[cnt]<=ans+cnt-1) cnt++;
		ans+=cnt-1;
	}
    cout<<ans<<'\n';
	return 0;
}