HDU_1698 Just a Hook(线段树)

　/*第二次做这题，其中有些坎坷，第一次做时理解的也不深刻，导致这次做起来有些费
劲。开始思路是一边更新一边计算总的Hook值，可惜一直WA，做了大半下午，晕死，先
更新完再统计一遍的方法过了，再后来上网查了一下，正好有个大牛也是用第一种思路做
的，参考了一下他的代码，发现原来我的flag初值整错了。。。。*/

//第一种思路代码（609+ms）：

#include <iostream>
#include <cstdio>
#define L(t) t << 1
#define R(t) t << 1 | 1

using namespace std;

const int N = 200005;

struct node
{
int l, r;
int flag;    //标记更新后的hook类型
int hook;    //统计总的hook值
}node[N*4];

void creat(int t, int l, int r)
{
node[t].l = l;
node[t].r = r;
node[t].flag = 0;
if(l == r)
{
node[t].hook = 1;
return;
}
int mid = (l + r) >> 1;

creat(L(t), l, mid);
creat(R(t), mid+1, r);

node[t].hook = node[L(t)].hook + node[R(t)].hook;
}

void updata(int t, int l, int r, int val)
{
if(node[t].l >= l && node[t].r <= r)
{
node[t].hook = (r-l+1)*val;
node[t].flag = val;
return ;
}

int mid = (node[t].l + node[t].r) >> 1;

if(node[t].flag)    //下分flag值和hook值
{
node[L(t)].flag = node[t].flag;
node[L(t)].hook = (mid - node[t].l + 1) * node[t].flag;    //左孩子

node[R(t)].flag = node[t].flag;
node[R(t)].hook = (node[t].r - mid) * node[t].flag;    //右孩子
node[t].flag = 0;
}

if(l > mid)
updata(R(t), l, r, val);
else if(r <= mid)
updata(L(t), l, r, val);
else
{
updata(L(t), l, mid, val);
updata(R(t), mid+1, r, val);
}

node[t].hook = node[L(t)].hook + node[R(t)].hook;
}

int main()
{
//freopen("data.in", "r", stdin);

int cas = 0, n, m, T, x, y, z;
scanf("%d", &T);
while(T--)
{
scanf("%d", &n);
creat(1, 1, n);
scanf("%d", &m);
while(m--)
{
scanf("%d%d%d", &x, &y, &z);
updata(1, x, y, z);
}
printf("Case %d: The total value of the hook is %d.\n", ++cas, node[1].hook);
}
return 0;
}

//第二种思路代码（468+ms）：

#include <iostream>
#include <cstdio>
#define L(t) t << 1
#define R(t) t << 1 | 1

using namespace std;

const int N = 200005;

struct node
{
int l, r;
int hook;    //存放当前结点的hook类型
}node[N*4];

int sum;

void creat(int t, int l, int r)
{
node[t].l = l;
node[t].r = r;
node[t].hook = 1;
if(l == r)    return ;

int mid = (l + r) >> 1;
creat(L(t), l, mid);
creat(R(t), mid + 1, r);
}

void updata(int t, int l, int r, int v)
{
if(node[t].l >= l && node[t].r <= r)
{
node[t].hook = v;
return ;
}

if(node[t].hook != -1)
{
node[L(t)].hook = node[t].hook;
node[R(t)].hook = node[t].hook;
node[t].hook = -1;
}

int mid = (node[t].l + node[t].r) >> 1;
if(l > mid)
updata(R(t), l, r, v);
else if(r <= mid)
updata(L(t), l, r, v);
else
{
updata(L(t), l, mid, v);
updata(R(t), mid + 1, r, v);
}
}

void query(int t, int l, int r)    //最后统计
{
if(node[t].hook != -1)
{
sum += node[t].hook * (r - l + 1);
return ;
}
int mid = (node[t].l + node[t].r) >> 1;

query(L(t), l, mid);
query(R(t), mid + 1, r);
}

int main()
{
//freopen("data.in", "r", stdin);

int T, n, m, x, y, z, cas = 0;
scanf("%d", &T);
while(T--)
{
scanf("%d", &n);
creat(1, 1, n);
scanf("%d", &m);
while(m--)
{
scanf("%d%d%d", &x, &y, &z);
updata(1, x, y, z);
}
sum = 0;
query(1, 1, n);
printf("Case %d: The total value of the hook is %d.\n", ++cas, sum);
}
return 0;
}

/*按说第一种思路应该比第二种快的，不知道为什么是这个情况*/