分块
时间复杂度取决于块的大小,一般块的大小为 n \sqrt{n} n,m次操作,则时间复杂度为 O ( m n ) O(m\sqrt{n}) O(mn)
预处理
先将序列分块,每块长度为 s q r t ( n ) sqrt(n) sqrt(n) 。最后一块需要特判一下。将每个块都标记左右两个端点 l [ i ] l[i] l[i] r [ i ] r[i] r[i] ,最后一块需要特判一下。
t = sqrt(n * 1.0);//参数类型只有double和float
int num = n / t;
if(n % t) num ++;
for(int i = 1; i <= num; i ++ )
ll[i] = (i - 1) * t + 1, rr[i] = i * t;
r[num] = n;
之后再用pos数组记录序列中每个元素所在的块,sum来记录每个块的值
for(int i = 1; i <= num; i ++ )
for(int j = ll[i]; j <= rr[i]; j ++)
pos[j] = i, sum[i] += a[j];
区间操作及查询
(1) 判断操作或查询的区间是否在同一块内
(2) 若在同一个块内,直接暴力操作
(3) 若不在同一个块中,先将最左边和最右边的两个块暴力操作,再将中间的块进行整体的操作,打上懒标记。
//区间更新,向a[l]~a[r]都加上数d
void change(int l, int r, int d)
{
int p = pos[l], q = pos[r];
if(p == q){
for(int i = l; i <= r; i ++ )
a[i] += d;
sum[i] += d * (r - l + 1);
}
else {
for(int i = l; i <= rr[p]; i ++)
a[i] += d;
sum[p] += d * (rr[p] - l + 1);
for(int i = l[q]; i <= r; i ++)
a[i] += d;
sum[q] += d * (r - ll[q] + 1);
//左右块进行操作
for(int i = p + 1; i <= q - 1; i ++)
add[i] += d;
//中间块进行懒标记
}
}
//求a[l]~a[r]的和
ll ans(int l, int r)
{
int p = pos[l], q = pos[r];
ll ans = 0;
if(p == q){
for(int i = l; i <= r; i ++)
ans += a[i];
ans += add[p] * (r - l + 1);
}
else {
for(int i = l; i <= rr[p]; i ++)
ans += a[i];
ans += add[p] * (rr[p] - l + 1);
for(int i = ll[q]; i <= r; i ++)
ans += a[i];
ans += add[p] * (r - ll[q] + 1);
//左右块操作
for(int i = p + 1; i <= q - 1; i ++)
ans += sum[i] + add[i] * (rr[i] - ll[i] + 1);
}
return ans;
}
POJ3468 数字序列
数字序列为1 12 123 1234 12345…1234567891011…
求第pos位置上的数字
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N = 2e5 + 10;
long long a[N], sum[N];
//a[i]用来存储第i块的数字序列的长度
//sum[i]用来存储前i个块的数字序列的总长度
int main()
{
int t;
cin >> t;
for(int i = 1; i < N; i ++ )
{
a[i] = a[i - 1] + (int)log10((double)i) + 1;
//log10(i) + 1用来表示第i组数字列的长度
sum[i] = sum[i - 1] + a[i];
}
while(t -- )
{
int num, idx = 1;
cin >> num;
while(sum[idx] < num) idx ++;
//说明num在第idx个块中
int pos = num - sum[-- idx], ans = 0, len = 0;
//pos为idx块的从1开始的第pos个位置
while(len < pos)
{
ans ++;
len += (int)log10((double)ans) + 1;
//len为第i组数的长度
}
cout << ans / (int)pow(10.0, len - pos) % 10 << endl;
//len - pos为第ans组中pos位置后面多余的数字的位数,/ 先删去pos后多余的数字
//再取模得pos位置的数字
//以123... 1234为例,len在1234中4的位置,pos位于1234中的2,
//ans / pow(10, len - pos) = 12,在对12进行%可得最后一位数
}
return 0;
}
HDU超级马里奥
给出一个从0开始, 长度为n的数组,有m次询问,每次询问给出l,r,h,求a[l]~a[r]中小于等于h的元素的个数
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int a[N], temp[N];
int ll[N], rr[N];
int pos[N];
int find(int l, int r, int h)
{
int ans = 0;
int p = pos[l], q = pos[r];
if(p == q){
for(int i = l; i <= r; i ++ )
if(a[i] <= h) ans ++;
}
else {
for(int i = l; i <= rr[p]; i ++)
if(a[i] <= h) ans ++;
for(int i = ll[q]; i <= r; i ++)
if(a[i] <= h) ans ++;
for(int i = p + 1; i <= q - 1; i ++ )
ans += upper_bound(temp + ll[i], temp + rr[i] + 1, h) - temp - ll[i];
}
return ans;
}
int main()
{
int n, m;
cin >> n >>m;
for(int i = 1; i <= n; i ++ ) {cin >> a[i]; temp[i] = a[i];}
int t = sqrt(n);
int num = n / t;
if(n % t) num ++;
for(int i = 1; i <= num; i ++)
ll[i] = (i - 1) * t + 1, rr[i] = i * t;
rr[num] = n;
for(int i = 1; i <= n; i ++ )
for(int j = ll[i]; j <= rr[i]; j ++)
pos[j] = i;//pos[j] = (i - 1) / t + 1;
for(int i = 1; i <= num; i ++)
sort(temp + ll[i], temp + rr[i] + 1);
while(m -- )
{
int l, r, h;
cin >> l >> r >> h;
cout << find(l + 1, r + 1, h) << endl;
}
return 0;
}