洛谷3372[模板]線段樹1

題目描述

如題,已知一個數列,你需要進行下面兩種操作:

1.將某區間每一個數加上x

2.求出某區間每一個數的和

輸入輸出格式

輸入格式:
第一行包含兩個整數N、M,分別表示該數列數字的個數和操作的總個數。

第二行包含N個用空格分隔的整數,其中第i個數字表示數列第i項的初始值。

接下來M行每行包含3或4個整數,表示一個操作,具體如下:

操作1: 格式:1 x y k 含義:將區間[x,y]內每個數加上k

操作2: 格式:2 x y 含義:輸出區間[x,y]內每個數的和

輸出格式:
輸出包含若干行整數,即爲所有操作2的結果。

輸入輸出樣例

輸入樣例#1:
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
輸出樣例#1:
11
8
20
說明

時空限制:1000ms,128M

數據規模:

對於30%的數據:N<=8,M<=10

對於70%的數據:N<=1000,M<=10000

對於100%的數據:N<=100000,M<=100000

(數據已經過加強^_^,保證在int64/long long數據範圍內)


模板題. 因爲以前很少寫標記永久化, 這裏貼一份代碼.
PS: 這題在寫對拍的時候出現了一些問題. 總結一下經驗, 在對拍程序裏面儘量不要出現 數字 || 字母 ||下劃線 以外的符號, 否則很有可能出現BAT命令臺無法運行程序而又沒有報錯的情況.

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxN = (int)1e5;
long long tag[maxN << 2];
long long sum[maxN << 2];
void modify(int p, long long del, int curL, int curR, int L, int R)
{
    if (L <= curL && R >= curR)
    {
        tag[p] += del;
        sum[p] += (curR - curL + 1) * del;
        return;
    }
    int mid = (curL + curR) >> 1;
    if(L <= mid)
        modify(p << 1, del, curL, mid, L, R);
    if(R > mid)
        modify((p << 1) + 1, del, mid + 1, curR, L, R);
    sum[p] += (min(curR, R) - max(curL, L) + 1) * del;
}
long long query(int p, int curL, int curR, int L, int R)
{
    if(L <= curL && R >= curR)
        return sum[p];
    int mid = (curL + curR) >> 1;
    long long ret = 0;
    if(L <= mid)
        ret += query(p << 1, curL, mid, L, R);
    if(R > mid)
        ret += query((p << 1) + 1, mid + 1, curR, L, R);
    ret += (min(curR, R) - max(curL, L) + 1) * tag[p];
    return ret;
}
int main()
{
    freopen("L3372.in", "r", stdin);
    freopen("L3372.out", "w", stdout);
    ios::sync_with_stdio(false);
    int n, m;
    cin >> n >> m;
    memset(tag, 0, sizeof(tag));
    memset(sum, 0, sizeof(sum));
    for(int i = 1; i <= n; i ++)
    {
        long long x;
        cin >> x;
        modify(1, x, 1, n, i, i);
    }
    for(int i = 0; i < m; i ++)
    {
        int opt;
        cin >> opt;
        if(opt == 1)
        {
            long long L, R, del;
            cin >> L >> R >> del;
            modify(1, del, 1, n, L, R);
        }
        else if (opt == 2)
        {
            int L, R;
            cin >> L >> R;
            cout << query(1, 1, n, L, R) << endl;
        }
    }
}
發佈了57 篇原創文章 · 獲贊 0 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章