1521 一维战舰

1521 一维战舰

 

对于n个格子，可放置战舰个数为(n+1)/(a+1)，当我们指定一个点时，则减少(right-left)/(a+1)-(x-left)/(a+1)-(right-x)/(a+1)。

每做一次指定点，就做一次减法运算，直至总个数<k或指定m次完毕。

 

AC代码：

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

const int MAX=2e5+10;
int vis[MAX];

int main(){
    int n,k,a,m,x;
    cin>>n>>k>>a>>m;
    int flag=-1;
    int left,right;
    memset(vis,0,sizeof(vis));
    int num=(n+1)/(a+1);
    for(int i=0;i<m;i++){
        cin>>x;
        vis[x]=1;
        for(left=x-1;left>0&&vis[left]==0;left--);
        for(right=x+1;right<=n&&vis[right]==0;right++);
        num-=(right-left)/(a+1)-(x-left)/(a+1)-(right-x)/(a+1);
        if(num<k&&flag==-1){
            flag=i+1;
            break;
        }
    }
    cout<<flag<<endl;
    return 0;
}