題目描述

給出一個長為n的數列，以及n個操作，操作涉及區間加法，詢問區間內小於某個值x的元素個數。

輸入格式

第一行輸入一個數字 。

第二行輸入n個數字，第i個數字為ai，以空格隔開。

接下來輸入n行詢問，每行輸入四個數字opt,l,r,c，以空格隔開。

若opt=0，表示將位於l,r的之間的數字都加c。

若opt=1，表示詢問l,r中，小於c^2的數字的個數。

輸出格式

對於每次詢問，輸出一行一個數字表示答案。

樣例

樣例輸入

4

1 2 2 3

0 1 3 1

1 1 3 2

1 1 4 1

1 2 3 2

樣例輸出

3

0

2

和上一題類似，不過這個題線段樹就無能為力了。按照線段樹大段維護，區域性樸素的思想，我們建立兩個陣列a和b，其中a是在原陣列之上進行操作後的陣列，b陣列是a陣列的拷貝，但同時每一塊內的元素是有序的。對於修改，分為端點在一塊內/不在一塊內，邊角塊直接暴力，中間的塊利用add陣列（類似線段樹的lazy tag）進行標記。注意，邊角塊處理完後一定要對b陣列中邊角塊位置的數進行重新排序！查詢類似，注意在中間塊中直接對每段用lower_bound函式查詢即可。

需要注意的細節：排序函式sort(b + L[i], b + R[i] + 1);右邊別忘+1！坑了半天TAT

#include <bits/stdc++.h>
using namespace std;
int a[500005], b[500005], add[500005];  
int L[500005], R[500005];
int pos[500005];
int n, t;
inline int read(){   int s=0,w=1;   char ch=getchar();   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}   while
 
(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();   return s*w;}
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;
        for (int i = L[p]; i <= R[p]; i++) 
        {
            a[i] += add[p];
            b[i] 
 
= a[i];
        }
        add[p] = 0;
        sort(b + L[p], b + R[p] + 1);
    } else {
        for (int i = p + 1; i <= q - 1; i++) add[i] += d;
        for (int i = l; i <= R[p]; i++) a[i] += d;
        for (int i = L[p]; i <= R[p]; i++) 
        {
            a[i] += add[p];
            b[i] = a[i];
        }
        add[p] = 0;
        sort(b + L[p], b + R[p] + 1);
        for (int i = L[q]; i <= r; i++) a[i] += d;
        for (int i = L[q]; i <= R[q]; i++) 
        {
            a[i] += add[q];
            b[i] = a[i];
        }
        add[q] = 0;
        sort(b + L[q], b + R[q] + 1);
    }
}

int ask(int l, int r, int c) {
    int p = pos[l], q = pos[r], ans = 0;
    if (p == q) {
        for (int i = l; i <= r; i++)
            if (a[i] + add[p] < c)
                ans++;
        return ans;
    } else {
        for (int i = p + 1; i <= q - 1; i++) {
            int ppos = lower_bound(b + L[i], b + R[i] + 1, c - add[i]) - b;
            if(ppos != R[i] + 1) ans += ppos - L[i];
            else ans += R[i] - L[i] + 1;
        }
        for (int i = l; i <= R[p]; i++)
            if (a[i] + add[p] < c) ans++;
        for (int i = L[q]; i <= r; i++)
            if (a[i] + add[q] < c) ans++;
        return ans;
    }
}
int main() {
    cin >> n;
    memset(add, 0, sizeof(add));
    for (int i = 1; i <= n; i++) {
        a[i] = read();
        b[i] = a[i];
    }
    t = sqrt(n);
    for (int i = 1; i <= t; i++) {
        L[i] = (i - 1) * sqrt(n) + 1;
        R[i] = i * sqrt(n);
    }
    if (R[t] < n)
        t++, L[t] = R[t - 1] + 1, R[t] = n;
    for (int i = 1; i <= t; i++)
    {
        for (int j = L[i]; j <= R[i]; j++) {
            pos[j] = i;
        }
        sort(b + L[i], b + R[i] + 1);
    }
    for (int i = 1; i <= n; i++) {
        int opt, l, r, c;
        opt = read(), l = read(), r = read(), c = read();
        if (opt == 0) {
            change(l, r, c);
        } else {
            cout << ask(l, r, c * c) << endl;
        }
    }
}