One-Dimensional Battle Ships

One-Dimensional Battle Ships

Alice and Bob love playing one-dimensional battle ships. They play on the field in the form of a line consisting of n square cells (that is, on a 1 × n table).

At the beginning of the game Alice puts k ships on the field without telling their positions to Bob. Each ship looks as a 1 × a rectangle (that is, it occupies a sequence of a consecutive squares of the field). The ships cannot intersect and even touch each other.

After that Bob makes a sequence of “shots”. He names cells of the field and Alice either says that the cell is empty (“miss”), or that the cell belongs to some ship (“hit”).

But here’s the problem! Alice like to cheat. May be that is why she responds to each Bob’s move with a “miss”.

Help Bob catch Alice cheating — find Bob’s first move, such that after it you can be sure that Alice cheated.

Input

The first line of the input contains three integers: n, k and a (1 ≤ n, k, a ≤ 2·105) — the size of the field, the number of the ships and the size of each ship. It is guaranteed that the n, k and a are such that you can put k ships of size a on the field, so that no two ships intersect or touch each other.

The second line contains integer m (1 ≤ m ≤ n) — the number of Bob’s moves.

The third line contains m distinct integers x1, x2, …, x**m, where x**i is the number of the cell where Bob made the i-th shot. The cells are numbered from left to right from 1 to n.

Output

Print a single integer — the number of such Bob’s first move, after which you can be sure that Alice lied. Bob’s moves are numbered from 1 to m in the order the were made. If the sought move doesn’t exist, then print “-1”.

Example

Input

11 3 3
5
4 8 6 1 11

Output

3

Input

5 1 3
2
1 5

Output

-1

Input

5 1 3
1
3

Output

1

题目大意:

有一个1*n的矩形,Alice在上面放了k条船,Bob每次会指定一个格子,然后Alice会告诉Bob这个格子有没有船。

然而Alice每次都撒谎说这个格子没有船,请告诉Bob他什么时候第一次能确定Alice一定在撒谎。

解析:

每当Bob猜一个位置,就将这个1*n的矩形分割成了若干块,同时可以确定的是,在Bob猜的位置是肯定没有战舰的,那么只有在Bob猜过的位置之间,以及开头和结尾的地方可能存在战舰,我们根据这几个空挡,去计算空挡内能够存在多少战舰,可以使用公式来计算(公式推导比较简单,这里不赘述了,自己画图试几次就好了)若当前情况下,最多能存在的战舰数量小于一开始说好的,那么肯定可以确定Alice撒谎了。

优化问题:

我们当然可以从1依次枚举到m,看看到底要多少次才能发现Alice在撒谎。

但是由于答案的线性特点,我们可以使用二分答案的方法来优化时间复杂度

代码:

#include<iostream>
#include<algorithm>
#include<vector>
#define INF 0x7ffff
using namespace std;

vector<int> v;
int n ,k, a, m;
int ans = INF;

bool check(int x) {
    vector<int> nodes;
    nodes.push_back(0);
    for(int i = 1; i<= x; ++i) nodes.push_back(v[i]);
    nodes.push_back(n + 1);
    sort(nodes.begin(), nodes.end());
    int tot = 0;
    for(int i = 0; i < nodes.size() - 1; ++i) {
        int l = nodes[i], r = nodes[i + 1];
        tot += ((r - l) / (a + 1));
    }
    if(tot >= k) return true;
    return false;
}

int main()
{
    cin >> n >> k >> a >> m;
    v.resize(m + 1);
    for(int i = 1; i <= m; ++i) {
        cin >> v[i];
    }
    int l = 1, r = m;
    while(l <= r) {
        int mid = (l + r) / 2;
        if(check(mid)) {
            l = mid + 1;
        } else {
            ans = min(ans, mid);
            r = mid - 1;
        }
    }
    if(ans == INF)  ans = -1;
    cout << ans << endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Here is an example code for a one-dimensional convolutional wavelet neural network using PyTorch: ```python import torch import torch.nn as nn import pywt class ConvWaveletNet(nn.Module): def __init__(self, num_classes): super(ConvWaveletNet, self).__init__() self.conv1 = nn.Conv1d(1, 16, kernel_size=3, stride=1, padding=1) self.relu1 = nn.ReLU() self.pool1 = nn.MaxPool1d(kernel_size=2, stride=2) self.conv2 = nn.Conv1d(16, 32, kernel_size=3, stride=1, padding=1) self.relu2 = nn.ReLU() self.pool2 = nn.MaxPool1d(kernel_size=2, stride=2) self.conv3 = nn.Conv1d(32, 64, kernel_size=3, stride=1, padding=1) self.relu3 = nn.ReLU() self.pool3 = nn.MaxPool1d(kernel_size=2, stride=2) self.fc1 = nn.Linear(64 * 4, 128) self.relu4 = nn.ReLU() self.fc2 = nn.Linear(128, num_classes) def forward(self, x): # Apply wavelet transform to the input signal cA, cD = pywt.dwt(x, 'db1') x = cA + cD x = torch.tensor(x).unsqueeze(0).unsqueeze(0).float() # add batch and channel dimensions # Convolutional layers x = self.conv1(x) x = self.relu1(x) x = self.pool1(x) x = self.conv2(x) x = self.relu2(x) x = self.pool2(x) x = self.conv3(x) x = self.relu3(x) x = self.pool3(x) # Fully connected layers x = x.view(-1, 64 * 4) x = self.fc1(x) x = self.relu4(x) x = self.fc2(x) return x ``` This network consists of three convolutional layers followed by two fully connected layers. The input signal is first transformed using the discrete wavelet transform, and then passed through the convolutional layers. The output of the last convolutional layer is flattened and passed through the fully connected layers to produce the final classification result. Note that this implementation uses the 'db1' wavelet for the wavelet transform, but other wavelets can also be used.

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值