cf1354D,多种解法

cf1354D 1900的题
题意:给你一个数组大小为n且其中的数的大小是恒小于等于n,然后对这个数组进行t个操作,如果操作数q<0则删除这个数组的第q小的数,如果q>0则把q插入这个数组中,保证|q|<=n,且不会越界,求这个数组中还存在的数是多少,打印任意一个即可,如果不存在则打印0
思路:有挺多方法的,树状数组优化,或者一种很巧妙的方法即假设一个x(1<=x<=n),然后判断x时候能存在于这个数组中即可,用二分来优化,具体实现可参考代码

#include 
#include 
#include 
#include 
#include 
using namespace std;
const int maxn = 1e6+10;
int n, t, a[maxn], q[maxn];
bool solve(int x){    //进来的元素为x
    int cnt = 0;    //cnt表示这个元素之下是否存在
    for(int i = 1; i <= n; i++){
        if(a[i]<=x) cnt++;  //在a[i]中进行遍历,如果小于x则cnt++,也就是说有cnt个数是小于x的
    }
    for(int i = 1; i <= t; i++){        //然后再对q[i],进行遍历
        //这里关于为什么一个是和cnt比较,一个是和x比较,
        //因为如果q[i]<0的话是删去第i小的数,所以是和cnt比较,
        //而如果q[i]>0的话是插入q[i],而不是在第q[i]位置插入一个数,所以是和x比较
        if(q[i]<0){         //如果q[i]<0,即表示要减去一个数
            if(-q[i]<=cnt) cnt--;   //对cnt进行判断,如果-q[i]>cnt的话就不会影响到x在a[i]中的位置
        }else{
            if(q[i]<=x) cnt++;      //这里同理,如果q[i]>x的话就不会影响x在a[i]中的位置
        }
    }
    return cnt>0;
}
int main(){
    scanf("%d%d", &n, &t);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
    }
    for(int i = 1; i <= t; i++){
        scanf("%d", &q[i]);
    }
    int l = 1, r = n+1;
    while(l < r){   //对所有元素进行二分
        int mid = (l+r)/2;
        if(solve(mid)) r = mid;     //如果返回的是true,即表示mid以下的数是存在的,所以r=mid
        else l = mid+1;             //反之则表示mid之下没有数存在,l=mid+1
    }
    printf("%d\n", l<=n?l:0);
    return 0;
}

树状数组优化
也就是在求排名那里进行了优化,二分

#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define per(i, a, n) for(int i = n-1; i >= a; i--)
#define INF 1ll<<60
const int maxn = 1e6+10;
int n, m;
int c[maxn], a[maxn], q[maxn];
int lowbit(int x){
    return x&(-x);
}
void add(int x, int i){
    while(x<=n){
        c[x] += i;
        x += lowbit(x);
    }
}
int query(int x){
    int res = 0;
    while(x){
        res += c[x];
        x -= lowbit(x);
    }
    return res;
}
int solve(int x){		//用了二分来求排名
    int l = 1, r = n;
    while(l<r){
        int mid = (l+r)/2;
        if(query(mid) >= x) r = mid;
        else l = mid+1;
    }
    return l;
}
int main(){
    scanf("%d%d", &n, &m);
    rep(i,1,n+1){
        scanf("%d", &a[i]);
        add(a[i], 1);
    }
    rep(i,1,m+1){
        scanf("%d", &q[i]);
    }
    int cnt = n;
    rep(i,1,m+1){
        if(q[i]>0){
            add(q[i], 1);cnt++;
        }else{
            add(solve(-q[i]), -1);
            cnt--;
        }
    }
    if(cnt>0){
        printf("%d\n", solve(1));
    }else printf("0\n");
    return 0;
}

你可能感兴趣的:(cf1354D,多种解法)