UVA12086 - Potentiometer(线段树/树状数组)

UVA12086 - Potentiometer(线段树/树状数组)

题目链接

题目大意:给你N个数字,然后有q个操作,操作类型:M代表修改某个位置的值为r,S代表查询某一段的数字和。

解题思路:线段树或者树状数组。

线段树

#include <cstdio>
#include <cstring>

const int N = 8e5 + 5;

int v[N];
int n;

int Query (int o, int l, int r, int ql, int qr) {

    int    m = l + (r - l) / 2;

    if (ql == l && r == qr)
        return v[o]; 

    if (qr <= m)
        return Query(2 * o, l, m, ql, qr);
    else if (ql > m)
        return Query(2 * o + 1, m + 1, r, ql, qr);
    else 
        return Query(2 * o, l, m, ql, m) + Query(2 * o + 1, m + 1, r, m + 1, qr);
}

void Update (int o, int l, int r, int p, int val) {

    int m = l + (r - l) / 2;

    if (l == r) 
        v[o] = val;
    else {

        if (p <= m)
            Update (2 * o, l, m, p, val);
        else
            Update (2 * o + 1, m + 1, r, p, val);

        v[o] = v[2 * o] + v[2 * o + 1];
    }
}

void solve () {

    char str[10];
    int x, y, r;
    while (scanf ("%s", str) && str[0] != 'E') {

        if (str[0] == 'M') {

            scanf ("%d%d", &x, &y);
            printf ("%d\n", Query (1, 1, n, x, y));
        } else {

            scanf ("%d%d", &x, &r);
            Update (1, 1, n, x, r);    
        }
    }
}

int main () {

    int cas = 0;
    int num;
    while (scanf ("%d", &n) && n) {

        if (cas)
            printf ("\n");
        printf ("Case %d:\n", ++cas);

        memset (v, 0, sizeof (v));
        for (int i = 1; i <= n; i++) {

            scanf ("%d", &num);
            Update (1, 1, n, i, num);
        }

        solve();            
    }
    return 0;
}

树状数组

#include <cstdio>
#include <cstring>

const int maxn = 2e5 + 5;

int lowbit (int x) { return x&-x; }

int n;
int A[maxn], C[maxn];

void Add (int x, int d) {

    while (x <= n) {

        C[x] += d;
        x += lowbit(x);
    }
}

int Sum (int x) {

    int ret = 0;
    while (x > 0) {

        ret += C[x];
        x -= lowbit(x);
    }
    return ret;
}

void solve () {

    char str[10];
    int x, r, y;
    while (scanf ("%s", str) && str[0] != 'E') {

        if (str[0] == 'M') {

            scanf ("%d%d", &x, &y);
            printf ("%d\n", Sum (y) - Sum (x - 1));
        } else {

            scanf ("%d%d", &x, &r);
            Add(x, r - A[x]);
            A[x] = r;//这个地方要记得修改
        }
    }
}

int main () {

    int cas = 0;
    while (scanf ("%d", &n) && n) {

        if (cas)
            printf ("\n");
        memset (C, 0, sizeof (C));
        for (int i = 1; i <= n; i++) {
            scanf ("%d", &A[i]);
            Add(i, A[i]);
        }
        printf ("Case %d:\n", ++cas);
        solve();
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值