题目链接:Just a Hook

分析:开始学习线段树了,好激动!改了好半天,终于A了。今天写了这个线段树,感觉又有新的体会了。主要是修改和查找。

修改函数Update():在修改区间的时候,如果正好和区间对应,则改变当前点的cover值,如果不对应,则要修改的区间必是当前区间的子区间,再当前区间的两个子区间修改就行了,关键点是:先将当前区间cover值赋给它的两个子区间,同时当前区间的cover值要赋零,表示当前区间里的保存的值不是同一个值。

查找函数Search():查找到当前区间cover值不为0时,就可以返回了,返回seg_tree[id].cover * (r - l),否则继续查找。

 

代码
#include<stdio.h>
#include
<stdlib.h>
#define NN 100000
struct node{
int l, r, cover;
}seg_tree[NN
* 8];


void Init(int l, int r, int id){
seg_tree[id].l
= l;
seg_tree[id].r
= r;
seg_tree[id].cover
= 1;
if (r - l <= 1){
return;
}

int mid = (l + r)>> 1;
Init(l, mid, id
* 2);
Init(mid, r, id
* 2 + 1);
}

void Update(int l, int r, int key, int id){
if (seg_tree[id].l == l && seg_tree[id].r == r){
seg_tree[id].cover
= key;
return ;
}
/*这个地方关键,要将当前值分别赋给它的两个子区间,自己置零*/
if (seg_tree[id].cover != 0){
seg_tree[id
* 2].cover = seg_tree[id].cover;
seg_tree[id
* 2 + 1].cover = seg_tree[id].cover;
seg_tree[id].cover
= 0;
}

int mid = (seg_tree[id].l + seg_tree[id].r) >> 1;

if (r <= mid){
Update(l, r, key, id
* 2);
}
else if (l >= mid){
Update(l, r, key, id
* 2 + 1);
}
else{
Update(l, mid, key, id
* 2);
Update(mid, r, key, id
* 2 + 1);
}
}

int Search(int l, int r, int id){
/*这个地方也是关键*/
if (seg_tree[id].cover > 0){
return seg_tree[id].cover * (r - l);
}

int mid = (seg_tree[id].l + seg_tree[id].r) >> 1;

if (r <= mid){
return Search(l, r, id * 2);
}
else if (l >= mid){
return Search(l, r, id * 2 + 1);
}
else{
return Search(l, mid, id * 2) + Search(mid, r, id * 2 + 1);
}
}
int main()
{
int T, it, ans, Q, N, x, y, z;
scanf(
"%d", &T);
for (it = 1; it <= T; it++){
scanf(
"%d%d", &N, &Q);
Init(
0, N, 1);
while (Q--){
scanf(
"%d%d%d", &x, &y, &z);
Update(x
- 1, y, z, 1); /*区间[x-1,x]表示点x*/
}

ans
= Search(0, N, 1);
printf(
"Case %d: The total value of the hook is %d.\n", it, ans);
}

// system("pause");
return 0;
}

 

posted on 2010-07-16 22:09  ylfdrib  阅读(526)  评论(0编辑  收藏  举报