codeforces-567-D: One-Dimensional Battle Ships（Set | 权值线段树）

题目：http://codeforces.com/problemset/problem/567/D

题意：n个位置，k条船，每条占a个连续位置，任意两条船不能重叠或相邻，每次破坏一个位置，问破坏几次后放不下k条船。

思路：对于i条船，至少需要 i*k + i-1 个连续的位置才可以放下，对于一段长len的连续位置有等式 i*k + i-1 <= len，至多可以放(len+1)/(a+1)条船。每次破坏位置x，找到x左边破坏的第一个位置 l ，右边被破坏的第一个位置 r ，每次减去区间贡献r-l+1+1)/(a+1)，再加上破坏后的贡献(x-l+1)/(a+1)、(r-x+1)/(a+1)即可。

代码：

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

int n, k, a, x, q;
set<int>s;
int main()
{
    cin >> n >> k >> a >> q;
    s.insert(0), s.insert(n+1);
    int num = (n+1)/(a+1), ans = -1;
    for(int i=1; i<=q; i++){
        cin >> x;
        if(ans!=-1) continue;
        auto it = s.lower_bound(x);
        int r = *it - 1, l = *(--it) + 1;
        num = num - (r-l+2)/(a+1) + (x-l+1)/(a+1) + (r-x+1)/(a+1);
        if(num < k) ans = i;
        s.insert(x);
    }
    cout << ans << endl;
}

也可以用权值线段树求x的前驱和后继

#include<bits/stdc++.h>
#define ls rt<<1
#define rs rt<<1|1
using namespace std;
const int maxn = 2e5+5;

int n, k, a, x, q;
int T[maxn<<2];
void update(int rt, int l, int r, int pos){
    T[rt] += 1;
    if(l == r) {return ;}
    int mid = l+r >> 1;
    if(pos <= mid) update(ls, l, mid, pos);
    else update(rs, mid+1, r, pos);
}
int Rank(int rt, int l, int r, int x){ // rank of x
    if(l == r) return 1; // 此处是数字1
    int mid = l+r >> 1;
    if(x <= mid) return Rank(ls, l, mid, x);
    else return T[ls] + Rank(rs, mid+1, r, x);
}
int query(int rt, int l, int r, int k){ // number whose rank is k
    if(l == r) return l;
    int mid = l+r >> 1;
    if(k <= T[ls]) return query(ls, l, mid, k);
    else return query(rs, mid+1, r, k-T[ls]);
}
int main()
{
    cin >> n >> k >> a >> q;
    int num = (n+1)/(a+1), ans = -1;
    for(int i=1; i<=q; i++){
        cin >> x;
        if(ans!=-1) continue;
        update(1, 1, n, x);
        int tmp = Rank(1, 1, n, x);
        int l= tmp==1 ? 1 :query(1, 1, n, tmp-1)+1, r= tmp==T[1] ? n :query(1, 1, n, tmp+1)-1;
        //cout << l << "--" << r << endl;
        num = num - (r-l+2)/(a+1) + (x-l+1)/(a+1) + (r-x+1)/(a+1);
        if(num < k) ans = i;
    }
    cout << ans << endl;
}