数列分块入门 3

思路:简单分块,写的时候一定要谨慎!!!,把y写错成x,wa到怀疑人生.......

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstring>
#include <stdio.h>
#include <algorithm>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <vector>
#include <cmath>
#include <stack>
#include <random>
using namespace std;
#define io ios::sync_with_stdio(0),cin.tie(0)
#define ms(arr) memset(arr,0,sizeof(arr))
#define LD long double
#define LL long long
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define inf 1<<30
#define  ull unsigned long long
const int Mod = 998244353;
const int maxn = 1e6 + 5;
int read() {
    int x = 0, f = 1;
    char ch = getchar();

    while (ch < '0' || ch > '9') {
        if (ch == '-')
            f = -1;

        ch = getchar();
    }

    while (ch >= '0' && ch <= '9') {
        x = x * 10 + ch - '0';
        ch = getchar();
    }

    return x * f;
}
int n, opt, l, r, c, block;
int belong[maxn], change[maxn];
int a[maxn];
vector<int>vt[5000];
void build() {
    block = sqrt(n);//分块操作,每一个块的大小为sqrt(n)

    for (int i = 1; i <= n; i++) {      //遍历n个数
        belong[i] = (i - 1) / block + 1;//确定第i个数属于第几个块
        vt[belong[i]].push_back(a[i]);//进入队列
    }

    for (int i = 1; i <= belong[n]; i++) {//对每一块进行排序
        sort(vt[i].begin(), vt[i].end());
    }
}
void rest(int x) {
    vt[x].clear();//清空这块

    for (int i = block * (x - 1) + 1; i <= min(block * x, n); i++) {//重新遍历这块
        vt[x].push_back(a[i]);
    }

    sort(vt[x].begin(),
         vt[x].end());//排序....坑点如果不在这里排序而是在查询的时候排序会因为查询次数过多而超时
}
void add(int x, int y, int val) {
    for (int i = x; i <= min(block * belong[x], y); i++) {
        a[i] += val;
    }

    rest(belong[x]);//重排

    if (belong[x] != belong[y]) {//不在同一块
        for (int i = (belong[y] - 1) * block + 1; i <= y; i++) {//对y所在块遍历[左边界,y]
            a[i] += val;
        }

        rest(belong[y]);

        for (int i = belong[x] + 1; i < belong[y]; i++) {//遍历[x所在的块+1,y所在的块-1],即中间块
            change[i] += val;
        }
    }
}
void query(int x, int y, int k) {
    int ans = -1;

    for (int i = x; i <= min(block * belong[x], y); i++) {//遍历找符合条件的数
        if (a[i] + change[belong[x]] < k)
            ans = max(ans, a[i] + change[belong[x]]);
    }

    if (belong[x] != belong[y]) {//不在同一块
        for (int i = (block * (belong[y] - 1) + 1); i <= y; i++) {//找y块的
            if (a[i] + change[belong[y]] < k)
                ans = max(ans, a[i] + change[belong[y]]);
        }

        for (int i = belong[x] + 1; i < belong[y]; i++) {//找x,y中间块
            int f = k - change[i];
            int t = lower_bound(vt[i].begin(), vt[i].end(), f) - vt[i].begin();

            //之前已经排序好了,不必再排
            if (t != 0)
                ans = max(ans, vt[i][t - 1] + change[i]);
        }
    }

    printf("%d\n", ans);
}

int main() {
    io;
    n = read();

    for (int i = 1; i <= n; i++) {
        a[i] = read();
    }

    build();//分块

    for (int i = 1; i <= n; i++) {
        opt = read();
        l = read();
        r = read();
        c = read();

        if (opt == 0) {
            add(l, r, c);
        } else {
            query(l, r, c);
        }
    }

    return 0;
}

 

posted @ 2021-03-22 20:19  夜灯长明  阅读(139)  评论(0编辑  收藏  举报