帕吉的肉钩

题意

在DotA游戏中，帕吉的肉钩是很多英雄最害怕的东西。钩子由连续若干段的等长金属棒制成。
现在帕吉对钩子由一些操作：
我们将金属棒1～n依次编号，帕吉可以把编号x～y的金属棒变成铜棒、银棒、金棒。
每段铜棒的价值是1；每段银棒的价值是2；每段金棒的价值是3。
肉钩的总价值是n段金属棒价值之和。帕吉想知道若干操作以后钩子的总价值。

输入格式

第一行一个整数N（1≤N≤105），表示肉钩金属棒的数量，初始状态为铜棒。
第二行一个整数Q（1≤Q≤105），表示操作数。
接下来Q行，一行三个整数，X，Y，Z（1≤X≤Y≤N，1≤Z≤3）。

输出格式

一行一个整数，表示钩子的总价值，用样例中的格式。

分析

一开始以为单纯把输入输出改改就行了，结果发现我太笨了。
这个题是把整个区间的值都换掉了，所以之前在modify函数中写的“+=”要改为“=”。
至于输出，我觉得直接输出s【1】的值也是可以的。

代码

#include<bits/stdc++.h>
using namespace std;
const int MAX_N=1e5+5;
int s[4*MAX_N],col[4*MAX_N];
void up(int p)
{
    s[p]=s[p*2]+s[p*2+1];
}
void down(int p,int l,int r)
{
    if(col[p])
    {
        int mid=(l+r)/2;
        s[p*2]=col[p]*(mid-l+1);
        s[p*2+1]=col[p]*(r-mid);
        col[p*2]=col[p];
        col[p*2+1]=col[p];
        col[p]=0;
    }
}
void modify(int p,int l,int r,int x,int y,int c)
{
    if(x<=l&&r<=y)
    {
        s[p]=(r-l+1)*c;
        col[p]=c;
        return;
    }
    down(p,l,r);
    int mid=(l+r)/2;
    if(x<=mid)
    {
        modify(p*2,l,mid,x,y,c);
    }
    if(y>mid)
    {
        modify(p*2+1,mid+1,r,x,y,c);
    }
    up(p);
}
int query(int p,int l,int r,int x,int y)
{
    if(x<=l&&r<=y)
    {
        return s[p];
    }
    down(p,l,r);
    int mid=(l+r)/2,res=0;
    if(x<=mid)
    {
        res+=query(p*2,l,mid,x,y);
    }
    if(y>mid)
    {
        res+=query(p*2+1,mid+1,r,x,y);
    }
    return res;
}
int main() {
	int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        modify(1,1,n,i,i,1);
    }
    int q;
    cin>>q;
    while(q--)
    {
        int x,y,z;
        cin>>x>>y>>z;
        modify(1,1,n,x,y,z);
    }
    printf("The total value of the hook is %d.\n",s[1]);
    return 0;
}