根据大神博客NotOnlySuccess的讲解,这个题目应该是属于线段树应用的第二阶段,也就是“成段更新”。大神博客已经有代码了,可能由于本题目不是很难所以没有过多的注释,如果是新手还是不容易看懂。所以这里就在啰嗦讲解一点吧。
题目:hdu 1698 Just a Hook
题目大意:dota中的屠夫(英雄)有一个钩子,钩子是连续的,现在假设钩子编号是从1到n,每个钩子有三种状态,分别对应的值是1、2、3(初始值都是1)。现在有m组操作:x,y,change;表示下标从x到y的钩子的值变为change。当然change的值只能是1、2、3。最后问这n个钩子的值的和sum是多少。
解题思路:首先建立线段树,每个断点至少记录一下当前这个区间的值,假设用key来记录,那么如果是1或者2或者3,就表示着段区间的所以值都是key。如果这段区间的值不一样那么就把key赋为-1。所以初始化应该都是1。对于怎么求sum,可以每次都updata_up向上去更新,这样就符合大多数线段树的思想了(对应代码1),但是对于这个题目,没有必要每次都向上去更新,可以等到所以操作完成后去遍历一下整个树就可以了(对应代码2)。
具体怎么实现可以参考一下代码:
hint:根据前辈们的教诲,程序最好写的整齐一些,这个不仅是指代码字母的整齐,还有布局和逻辑思想的整齐。
代码1:
//标准版 #include <iostream> #include <cstdio> using namespace std; typedef struct RMQ{ struct RMQ *lc, *rc; int left, right, key, sum; RMQ(){ key=1; } }*rmq; rmq great(int x, int y){ rmq q = new RMQ; q->left = x; q->right = y; q->sum = y - x + 1; if(x == y){ q->lc = q->rc = NULL; return q; } q->lc = great(x,(x+y)>>1); q->rc = great(((x+y)>>1)+1,y); return q; } void update_down(rmq q){ if(q ->key != -1){ q ->lc ->key = q ->rc ->key =q ->key; q ->lc ->sum = q ->lc ->key * (q ->lc ->right - q ->lc ->left + 1); q ->rc ->sum = q ->rc ->key * (q ->rc ->right - q ->rc ->left + 1); q ->key = -1; } } void update_up(rmq q){ q ->sum = q ->lc ->sum + q ->rc ->sum; } void update(rmq q, int x, int y, int change){ if(x <= q->left && y>= q->right){ q->key = change; q->sum = change * (q ->right - q ->left + 1); return; } int mid = (q ->left + q->right)>>1; update_down(q); if(x <=mid ) update(q->lc, x, y, change); if(y > mid) update (q->rc, x, y, change); update_up(q); } int main(){ int T, cas=1; cin >> T; while( T -- ) { int n, m, x, y, change; cin >> n >>m; rmq head = new RMQ; head = great(1, n); head ->sum = n; for(int i = 0; i < m; i ++){ cin >> x >> y >> change; update(head,x, y, change); } printf("Case %d: The total value of the hook is %d.\n",cas++,head ->sum); } }
//简洁版 #include <iostream> #include <cstdio> using namespace std; typedef struct RMQ{ struct RMQ *lc, *rc; int left, right, key, sum; RMQ(){ key = 1; } }*rmq; rmq great(int x,int y){ rmq q = new RMQ; q ->left = x; q ->right = y; if(x == y){ q ->lc = q ->rc = NULL; return q; } int mid = (x + y) >> 1; q ->lc = great(x, mid); q ->rc = great(mid + 1, y); return q; } void update_down(rmq q){ if(q ->key!=-1){ q ->lc ->key = q ->rc ->key = q ->key; q ->key = -1; } } void update(rmq q, int x, int y, int key){ if(x <= q ->left && y >= q ->right){ q ->key = key; return ; } update_down(q); int mid = (q ->left + q ->right) >> 1; if(x <= mid) update(q ->lc, x, y, key); if(y > mid) update(q ->rc, x, y, key); } int fun(rmq q){ if(q != NULL){ if(q ->key != -1) return q ->key * (q ->right - q ->left + 1); else return fun(q ->lc) + fun(q ->rc); } else return 0; } int main(){ int T,cas=1; cin >> T; while(T --){ int n, m, x, y, change; cin >> n >> m; rmq head = new RMQ; head = great(1,n); for(int i = 0; i < m; i ++){ cin >> x >> y >> change; update(head, x, y, change); } printf("Case %d: The total value of the hook is %d.\n",cas++,fun(head)); } }