ZUFEOJ 2147 07染色带谜题

2147: 07染色带谜题

时间限制: 1 Sec  内存限制: 128 MB
提交: 170  解决: 21
[提交][状态][讨论版][Edit] [TestData]

题目描述

现在给你一个长为N的染色带，有M种颜色，标号分别为0,1,2，…，M-1,染色带初始的颜色为标号0，现在对它进行K次操作，第i次操作，是对染色带的[Li,Ri]区间染上第Si种颜色（即把这段区间原来的颜色给覆盖掉），最后问你染色带进行这么多次操作后染色带上有几种颜色。

输入

输入包含多组数据，每组数据第一行包含3个正整数分别是N,M,K（0<N<=10000000，0<M<=10000，0<K<=10000）,它们用空格隔开，接下来K行分别是3个整数Li ，Ri，Si，分别满足0<=Li <Ri<=N, 0<=Si<M。

输出

输出对于每组数据输出一个数字代表染色带上最后有几种颜色。

样例输入

2 2 0
2 2 1
0 2 1
3 3 3
0 2 1
1 3 2
2 3 0

样例输出

1
1
3

首先吐槽一下题目，第一句，应该是$n+1$长度的吧。。。直接上了个线段树区间覆盖，写完一直$WA$，对拍去了，发现标程是错的，数据也是错的。搞笑的是这题是$2014$年学校校赛的题目，当时数据就是错的，但是现场有人$AC$。。现在我把数据改正确了，历史翻案。

#include<map>
#include<set>
#include<ctime>
#include<cmath>
#include<queue>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;

int n,m,k;
int L[],R[],s[];
int p[],sz;
int t[],y[];
set<int>r;

int get(int x)
{
    int left =,right =sz,res;
    while(left<=right)
    {
        int mid = (left + right)/;
        if(p[mid]>x) right = mid-;
        else if(p[mid]==x) res=mid,right = mid-;
        else left = mid+;
    }
    return res;
}

void pushDown(int rt)
{
    if(t[rt]==-) return ;
    t[*rt] = t[rt];
    t[*rt+] = t[rt];
    t[rt]=-;
}

void update(int LL,int RR,int col,int l,int r,int rt)
{
    if(LL<=l&&r<=RR)
    {
        t[rt]=col;
        return ;
    }

    int m = (l+r)/;
    pushDown(rt);
    if(LL<=m) update(LL,RR,col,l,m,*rt);
    if(RR>m) update(LL,RR,col,m+,r,*rt+);
}

void dfs(int l,int r,int rt)
{
    if(l==r)
    {
        y[t[rt]]=;
        return ;
    }
    int m = (l+r)/;
    pushDown(rt);
    dfs(l,m,*rt);
    dfs(m+,r,*rt+);
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        t[rt]=;
        return ;
    }

    t[rt]=-;
    int m = (l+r)/;
    build(l,m,*rt);
    build(m+,r,*rt+);
}

int main()
{
    //freopen("D:\\out.txt","w",stdout);
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        sz=;
        memset(t,,sizeof t); memset(y,,sizeof y); r.clear();

        for(int i=;i<=k;i++)
        {
            scanf("%d%d%d",&L[i],&R[i],&s[i]);
            if(r.count(L[i])==) sz++, p[sz]=L[i] , r.insert(L[i]);
            if(r.count(R[i])==) sz++, p[sz]=R[i] , r.insert(R[i]);

            if(r.count(L[i]-)==&&L[i]->=) sz++, p[sz]=L[i]- , r.insert(L[i]-);
            if(r.count(L[i]+)==&&L[i]+<=n) sz++, p[sz]=L[i]+ , r.insert(L[i]+);

            if(r.count(R[i]-)==&&R[i]->=) sz++, p[sz]=R[i]- , r.insert(R[i]-);
            if(r.count(R[i]+)==&&R[i]+<=n) sz++, p[sz]=R[i]+ , r.insert(R[i]+);
        }

        if(r.count()==) sz++, p[sz]= , r.insert();
        if(r.count(n)==) sz++, p[sz]=n , r.insert(n);

        sort(p+,p++sz);

        build(,sz,);
        for(int i=;i<=k;i++) L[i] = get(L[i]), R[i] = get(R[i]);
        for(int i=;i<=k;i++) update(L[i],R[i],s[i],,sz,);

        dfs(,sz,);
        int ans=;
        for(int i=;i<=;i++) ans=ans+y[i];
        printf("%d\n",ans);

    }
    return ;
}