poj 3468 A Simple Problem with Integers 树状数组 或 线段树

题目链接:http://poj.org/problem?id=3468

一般说来 树状数组是

【单点更新 区间查询】

而本题是“区间更新 区间查询”

所以要改成维护前缀和

推公式的过程见《挑战程序设计竞赛》第181~182面

代码中bit0是i的零次项

bit1是i的一次项

以后以此类推

在[l, r]加数的时候

写出公式

在l的地方 一次项及以上的直接写

然后在零次项那减去

在r的地方 一次项及以上的减掉之前加上的

然后再零次项那加上“公式化简之后的最终结果 减去 之前在零次项加的那一项”

【虽然很乱但我姑且是这么理解的】

以下是树状数组的写法:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <map>
#include <stack>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;

const ull B = 9973;
const int maxn = 110000;
const ll M = 23333;

ll bit0[maxn];
ll bit1[maxn];

int n, q;

ll sum(ll *b, int i)
{
    ll s = 0;
    while(i > 0)
    {
        s += b[i];
        i -= i & -i;
    }
    return s;
}

void add(ll *b, int i, int v)
{
    while(i <= n)
    {
        b[i] += v;
        i += i & -i;
    }
}

int main()
{
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    //freopen("output.txt", "w", stdout);
#endif

    while(scanf("%d%d", &n, &q) == 2)
    {
        for(int i = 1; i <= n; i++)
        {
            int t;
            scanf("%d", &t);
            add(bit0, i, t);
        }

        for(int i = 0; i < q; i++)
        {
            char mode;
            scanf("%C", &mode);
            scanf("%C", &mode);
            if(mode == ‘C‘)
            {
                int l, r, x;
                scanf("%d%d%d", &l, &r, &x);
                add(bit0, l, -x*(l-1));
                add(bit1, l, x);
                add(bit0, r+1, x*r);
                add(bit1, r+1, -x);
            }
            else
            {
                int l, r;
                scanf("%d%d", &l, &r);
                ll res = 0;
                res += sum(bit0, r) + sum(bit1, r)*r;
                res -= sum(bit0, l-1) + sum(bit1, l-1)*(l-1);
                printf("%lld\n", res);
            }
        }

    }

    return 0;
}

然后 再有一个线段树的写法:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <set>
#include <map>
#include <stack>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;

const int maxn = 410000;

ll data[maxn], datb[maxn];

int n, q;

void add(int a, int b, int x, int k, int l, int r)
{
    if(a <= l && b >= r)
    {
        data[k] += x;
    }
    else if(l < b && r > a)
    {
        datb[k] += (min(b, r) - max(a, l))*x;
        add(a, b, x, k*2+1, l, (l+r)/2);
        add(a, b, x, k*2+2, (l+r)/2, r);
    }
}

ll sum(int a, int b, int k, int l, int r)
{
    if(b <= l || a >= r)
        return 0;
    else if(a <= l && b >= r)
    {
        return data[k]*(r-l) + datb[k];
    }
    else
    {
        ll res = (min(b, r) - max(a, l))*data[k];
        res += sum(a, b, k*2+1, l, (l+r)/2);
        res += sum(a, b, k*2+2, (l+r)/2, r);
        return res;
    }
}

int main()
{
#ifdef LOCAL
    freopen("input.txt", "r", stdin);
    //freopen("output.txt", "w", stdout);
#endif

    while(scanf("%d%d", &n, &q) == 2)
    {

        for(int i = 0; i < n; i++)
        {
            int t;
            scanf("%d", &t);
            add(i, i+1, t, 0, 0, n);
        }

        for(int i = 0; i < q; i++)
        {
            char mode;
            scanf("%c", &mode);
            scanf("%c", &mode);

            if(mode == ‘C‘)
            {
                int a, b, x;
                scanf("%d%d%d", &a, &b, &x);
                add(a-1, b, x, 0, 0, n);
            }
            else
            {
                int a, b;
                scanf("%d%d", &a, &b);
                printf("%lld\n", sum(a-1, b, 0, 0, n));
            }
        }

    }

    return 0;
}
时间: 2024-11-05 16:05:29

poj 3468 A Simple Problem with Integers 树状数组 或 线段树的相关文章

POJ 3468 A Simple Problem with Integers(线段树)

题目链接:http://poj.org/problem?id=3468 A Simple Problem with Integers Time Limit: 5000MS   Memory Limit: 131072K Total Submissions: 56005   Accepted: 16903 Case Time Limit: 2000MS Description You have N integers, A1, A2, ... , AN. You need to deal with

poj 3468 A Simple Problem with Integers 【线段树-成段更新】

题目:poj 3468 A Simple Problem with Integers 题意:给出n个数,两种操作 1:l -- r 上的所有值加一个值val 2:求l---r 区间上的和 分析:线段树成段更新,成段求和 树中的每个点设两个变量sum 和 num ,分别保存区间 l--r 的和 和l---r 每个值要加的值 对于更新操作:对于要更新到的区间上面的区间,直接进行操作 加上 (r - l +1)* val .下面的区间标记num += val 对于求和操作,每次进行延迟更新,把num值

POJ 3468 A Simple Problem with Integers(线段树区间更新)

题目地址:POJ 3468 打了个篮球回来果然神经有点冲动..无脑的狂交了8次WA..居然是更新的时候把r-l写成了l-r... 这题就是区间更新裸题.区间更新就是加一个lazy标记,延迟标记,只有向下查询的时候才将lazy标记向下更新.其他的均按线段树的来就行. 代码如下: #include <iostream> #include <cstdio> #include <cstring> #include <math.h> #include <stac

POJ 3468 A Simple Problem with Integers 【树状数组】

题目链接:http://poj.org/problem?id=3468 题目大意:给出一组数组v[i],有两种操作,一种给出两个数a,b,要求输出v[a]到v[b]之间的和,另一种给出三个数a,b,c,让v[a]到v[b]之间的数全都加上c. 完全是树状数组能够实现的功能,但是如果就这样单纯的套用模板,做第二种操作是更新每个值,这样的操作就有可能超时. 换一种思路,既然第二种操作是给某区间上的所有数加上相同的值,那么应该是能够简化的才对. 假设数组sum[i]为原数组从v[1]到v[i]的和,数

poj 3468:A Simple Problem with Integers(线段树,区间修改求和)

A Simple Problem with Integers Time Limit: 5000MS   Memory Limit: 131072K Total Submissions: 58269   Accepted: 17753 Case Time Limit: 2000MS Description You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of

线段树(成段更新) POJ 3468 A Simple Problem with Integers

题目传送门 1 /* 2 线段树-成段更新:裸题,成段增减,区间求和 3 注意:开long long:) 4 */ 5 #include <cstdio> 6 #include <iostream> 7 #include <algorithm> 8 #include <cstring> 9 #include <cmath> 10 using namespace std; 11 12 #define lson l, mid, rt <<

poj 3468 A Simple Problem with Integers (线段树 成段更新 加值 求和)

题目链接 题意: 只有这两种操作 C a b c" means adding c to each of Aa, Aa+1, ... , Ab. -10000 ≤ c ≤ 10000."Q a b" means querying the sum of Aa, Aa+1, ... , Ab. 分析:自己写的有点麻烦了,写的时候手残+脑残,改了好久. val+lazy*(r-l+1)表示和,如果lazy==0表示当前区间加的值不统一. 1 #include <iostream

POJ 3468 A Simple Problem with Integers(线段树 区间更新)

Description You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval. In

POJ 3468 A Simple Problem with Integers(详细题解)

这是个线段树题目,做之前必须要有些线段树基础才行不然你是很难理解的. 此题的难点就是在于你加的数要怎么加,加入你一直加到叶子节点的话,复杂度势必会很高的 具体思路 在增加时,如果要加的区间正好覆盖一个节点,则增加其节点的Inc值,不再往下走,否则要更新Sum(加上本次增量),再将增量往下传. 这样更新的复杂度就是O(log(n))在查询时,如果待查区间不是正好覆盖一个节点,就将节点的Inc往下带,然后将Inc代表的所有增量累加到Sum上后将Inc清0,接下来再往下查询. Inc往下带的过程也是区

POJ - 3468 A Simple Problem with Integers (区间求和)

Description You have N integers, A1, A2, ... , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval. In