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