这道神奇的模拟题,带一点贪心,其实蛮水的,仔细思考就能ac。
首先我们模拟一下样例2
发现其实答案0 1 2 3也可以,仔细观察题目,我们发现了一句有意思的话:允许输出任意一组答案。
所以就可使用xjb算法,复杂度为O(玄学)(逃
我们假装有两个小D;一个从0开始跳;一个从n开始跳; 只要距离大于s且没被跳过就跳,之所以说是贪心是因为 我们尽可能让小D1号多跳一点石头,小D2号少跳一点(当然可以反过来),最后扫一遍,如果有石头没跳,就输出“NO”,反之就输出“YES”,然后再按小D1号再到小D2号的顺序输出,核心如下
1 for(int i=0;i<=n;i++) 2 if(i-pos>=s&&flag[i]==1) 3 {flag[i]=0;ans[now]=a[i];now++;pos=i;} 4 pos=n; 5 for(int i=n;i>=0;i--) 6 if(pos-i>=s&&flag[i]==1) 7 {flag[i]=0;ans[now]=a[i];now++;pos=i;}
值得一提的是,起点0和终点n也算石头,但是0最后输出,详见代码。
1 #include<bits/stdc++.h>//万能头文件,noip不能用 2 #define maxn 100010 3 using namespace std; 4 int a[maxn],ans[maxn];//a记录石头序号,ans记录跳的顺序(即跳的石头序号) 5 bool flag[maxn];//记录可不可以跳 6 inline void read(int &x){//快读 7 x=0;int f=1; 8 char ch=getchar(); 9 while(ch<'0'||ch>'9'){ 10 if(ch=='-')f=-1; 11 ch=getchar(); 12 } 13 while(ch>='0'&&ch<='9'){ 14 x=x*10+ch-'0'; 15 ch=getchar(); 16 }x*=f; 17 } 18 int main(){ 19 //memset(flag,0,sizeof(flag)); 20 int n,m,s,x,pos=n,now=1;//pos是当前位置,now是已跳石头个数 21 read(n);read(m);read(s); 22 for(int i=1;i<=m;++i) 23 {read(x);flag[x]=1;a[x]=i;}//flag==1代表可以跳 24 flag[0]=1;flag[n]=1; 25 a[0]=0;a[n]=m+1;//初始化,0和n也算石头 26 for(int i=0;i<=n;i++) 27 if(i-pos>=s&&flag[i]==1) 28 {flag[i]=0;ans[now]=a[i];now++;pos=i;} 29 pos=n; 30 for(int i=n;i>=0;i--) 31 if(pos-i>=s&&flag[i]==1) 32 {flag[i]=0;ans[now]=a[i];now++;pos=i;}//上已讲解 33 for(int i=0;i<=n;i++) 34 if(flag[i]==1) {printf("NO");return 0;}//有石头没跳 35 printf("YES "); 36 for(int i=1;i<=now-1;i++) 37 printf("%d ",ans[i]); 38 }