HDU 1698 Just a Hook

来源:互联网 发布:逍遥问道数据库修改器 编辑:程序博客网 时间:2024/04/29 17:01

新手写的线段树,看了定义就写,代码风格不好,请大神指教,题目意思就是简单维护区间和。。。第一篇博客,感觉有必要总结经验。。`

#include<iostream>using namespace std;const int size1 = 400000;//大概开到4倍int shu[size1];int lazy[size1];//懒惰标记,即是需要在对这区间的儿子们进行操作void build(int l, int r, int k)   //建树,通过递归去建立{    if (l == r)  //L,R是区间,很容易混淆,k只是数组存树的一个映射。所以下面基本都是折中操作,                       {               //否则K会出错,这里是到了叶子节点,就是我们的单个参数,自然建立后跳出        shu[k] = 1;        return;    }    int mid = (l + r) / 2;       build(l, mid, k * 2);//建立左子树    build(mid + 1, r, k * 2 + 1);//建立右子树    shu[k] = shu[2 * k + 1] + shu[2 * k];   //这里是维护线段树的保存信息,通过递归从底而上维护,从而保证了正确性}void change(int l, int r, int k, int num){    if (l == r)     //如果是叶子,懒惰标记到此为止    {        lazy[k] = 0;    }    else        lazy[k] = num;//懒惰标记需要下放    shu[k] = (r - l + 1)*num;//维护区间和}void change(int l, int r, int k, int x, int y, int num)//修改函数,由于第一次写,按照定义也没参考别人优秀代码,思维混乱导致代码不好。{    if (l == r)//叶子,修改后结束,等待自动回溯更新父节点信息    {        shu[k] = num;        return;    }    else//若不是叶子    {        if (l != r && (l <= x || r >= y) && lazy[k] != 0) //如果懒惰标记存在,且在操作区间内        {            int mid = (l + r) / 2;            change(l, mid, k * 2, lazy[k]);//下放懒惰标记的操作到儿子            change(mid + 1, r, k * 2 + 1, lazy[k]);            lazy[k] = 0;//取消懒惰标记        }        if (l == x&&r == y&&l != r || (x <= l&&r <= y))//如果操作的区间刚好为递归区间或者递归区间被操作区间包涵,下放操作到节点儿子        {            lazy[k] = num;            shu[k] = (r - l + 1)*num;            return;        }        else//否则,递归遍历左右儿子        {            int mid = (l + r) / 2;            if (x <= mid)//逼近原理,即如果操作区间在递归区间的中间的左边,则递归处理左儿子,这样肯定可以刚好递归到操作区间            {                change(l, mid, k * 2, x, y, num);            }            if (y>mid)//同上的右儿子版本            {                change(mid + 1, r, k * 2 + 1, x, y, num);            }        }    }    shu[k] = shu[2 * k + 1] + shu[2 * k];//回溯维护父节点信息}int main(){    int t;    cin >> t;    for (int i = 1; i <= t; i++)    {        memset(lazy, 0, sizeof(lazy));        memset(shu, 0, sizeof(shu));        int n;        cin >> n;        build(1, n, 1);        int q;        cin >> q;        for (int i = 0; i<q; i++)        {            int x, y, num;            scanf("%d%d%d", &x, &y, &num);            change(1, n, 1, x, y, num);        }        printf("%s%d%c%s%d%c\n", "Case ", i, ':', " The total value of the hook is ", shu[1],'.');//因为问整段区间和,访问根节点就好    }}
0 0