首页 > 代码库 > 1521 一维战舰

1521 一维战舰

1521 一维战舰

 

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

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

 

AC代码:

 1 #include<bits/stdc++.h> 2 using namespace std; 3  4 const int MAX=2e5+10; 5 int vis[MAX]; 6  7 int main(){ 8     int n,k,a,m,x; 9     cin>>n>>k>>a>>m;10     int flag=-1;11     int left,right;12     memset(vis,0,sizeof(vis));13     int num=(n+1)/(a+1);14     for(int i=0;i<m;i++){15         cin>>x;16         vis[x]=1;17         for(left=x-1;left>0&&vis[left]==0;left--);18         for(right=x+1;right<=n&&vis[right]==0;right++);19         num-=(right-left)/(a+1)-(x-left)/(a+1)-(right-x)/(a+1);20         if(num<k&&flag==-1){21             flag=i+1;22             break;23         }24     }25     cout<<flag<<endl;26     return 0;27 }

 

1521 一维战舰