1.分块算法 (n*sqrt(n)*log^2n) 3200+ms
给出n个数,有两种操作,一种是修改某个数的值,另一种查询指定区间第k大。
比较快的做法是树套树,而分块算法复杂度比较高写起来方便。分块算法可以很简单的处理单独修改某个值的情况。
将n个数分成num块,每块大小siz=n/num。每一个块内部进行排序,查询[l,r]第k大时,先二分答案,对于完全包含在区间的内块直接二分搜索,而对于区间两端只有部分包含的则直接遍历查找。复杂度是logn*(2*siz+num*log(siz)),这里取siz=sqrt(n),复杂度主要就是logn*sqrt(n)*logn。
对于修改,因为一个数只在一个块里,直接在那个块里进行修改,原本所在块是已排序的,修改后只有一个数不满足顺序,做一次冒泡就好了,复杂为siz=sqrt(n)。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>
using namespace std;
const int maxn=5e4+10;
int n,m;
int num,siz;
vector<int> sorted[1000];
int arr[maxn];
int main()
{
int X;
cin>>X;
while(X--){
cin>>n>>m;
for(int i=0; i<n; i++)
scanf("%d", arr+i);
siz=sqrt(n);//每一块的大小
num=(n+siz-1)/siz;//共num块
for(int i=0; i<num; i++) sorted[i].clear();
for(int i=0; i<n; i++){
sorted[i/siz].push_back(arr[i]);
}
for(int i=0; i<num; i++)
sort(sorted[i].begin(), sorted[i].end());
while(m--){
char s[5]; int a,b,k;
scanf("%s%d%d", s, &a, &b);
if(s[0]=='Q'){
scanf("%d", &k);
a--;b--;
int l=0,r=1e9+1;
int ans;
while(l<r){//二分答案统计<=m的有多少个数
int m=(l+r)>>1;
int cnt=0;
for(int i=a/siz+1; i<b/siz; i++)//处理一定完全包含的块
cnt+=upper_bound(sorted[i].begin(), sorted[i].end(), m)-sorted[i].begin();
if(a%siz==0)//处理左端可能部分包含的块
cnt+=upper_bound(sorted[a/siz].begin(), sorted[a/siz].end(), m)-sorted[a/siz].begin();
else{
for(int i=a; i%siz!=0; i++)
cnt+=arr[i]<=m;
}
if(b%siz==siz-1)//处理右端可能部分包含的块
cnt+=upper_bound(sorted[b/siz].begin(), sorted[b/siz].end(), m)-sorted[b/siz].begin();
else{
for(int i=b; i%siz!=siz-1; i--)
cnt+=arr[i]&l