hdu 1698 Just a Hook

初学线段树，参考了别人的做法，这里记录一下自己理解，线段树是一种思想，没有具体的模板

 

在这里，线段树每个节点有三个成员:{ml,mr,mcolor}分别表示 区间的头尾跟覆盖的颜色，

mcolor为0时，表示[ml,mr]这个区间内覆盖的颜色不一样

mcolor不为0时，表示[ml,mr]这个区间内覆盖的颜色一样

[s t]为要操作的线段

 

更新时：

当[s t]恰好落在区间上，就让这个区间的mcolor为我们指定的color， 否则，区间的mcolor为0，这个区间的状态转移到处理它的两个孩子的状态

 

查询时：

当[s,t]访问到的区间mcolor刚好为0时，那么就直接返回当前区间的mcolor*(r-l+1)，否则，访问它的子区间

#include <cstdio> #include <cstring> #define MN 100000 #define MAXN MN*3 struct Segtree{ public: // 运行同时初始化 void setsize(int size){ msize = size; ini(1, size, 1, 1); } void update(int s, int t, int ind, int color) { if (s==ml[ind] && t==mr[ind]) { mcolor[ind] = color; return; } if (mcolor[ind]) { // 关键 mcolor[ind*2] = mcolor[ind]; mcolor[ind*2+1] = mcolor[ind]; mcolor[ind] = 0; } int mid = (ml[ind]+mr[ind])/2; if (t <= mid) { update(s, t, ind*2, color); // * } else if (s > mid) { update(s, t, ind*2 + 1, color); } else { update(s, mid, ind*2, color); update(mid + 1, t, ind*2 + 1, color); } mcolor[ind] = 0; } int query(int s, int t, int ind) { //printf("s:%d,/tt:%d,/tind:%d,/tmcolor[]:%d/n", s, t, ind, mcolor[ind]); if (mcolor[ind]) { // 关键 return mcolor[ind]*(t - s + 1); } int mid = (ml[ind]+mr[ind])/2; if (t <= mid) { return query(s, t, ind*2); } else if (s > mid) { return query(s, t, ind*2 + 1); } else { return query(s, mid, ind*2) + query(mid+1, t, ind*2 + 1); } } private: int ml[MAXN],mr[MAXN],mcolor[MAXN],msize; void ini(int s, int t, int ind, int color) { // [s,t] ml[ind] = s; mr[ind] = t; mcolor[ind] = color; int mid = (s+t)/2; if (s==t) { mcolor[ind] = color; return; } ini(s, mid, ind*2, color); ini(mid+1, t, ind*2+1, color); } }; Segtree stree; int main () { int t,n; int m,le,ri,color; scanf("%d", &t); for (int i = 1; i <= t; ++i) { scanf("%d", &n); stree.setsize(n); scanf("%d", &m); for (int j = 0; j < m; ++j) { scanf("%d%d%d", &le, &ri, &color); stree.update(le, ri, 1, color); } int ans = stree.query(1, n, 1); printf("Case %d: The total value of the hook is %d./n", i, ans); } return 0; }