2018年湘潭大学程序设计竞赛 H 统计颜色【线段树+状态压缩】

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld

题目描述 

n个桶按顺序排列,我们用1~n给桶标号。有两种操作:
1 l r c 区间[l,r]中的每个桶中都放入一个颜色为c的球 (1≤l,r ≤n,l≤r,0≤c≤60)
2 l r   查询区间[l,r]的桶中有多少种不同颜色的球     (1≤l,r ≤n,l≤r)

输入描述:

有多组数据,对于每组数据:
第一行有两个整数n,m(1≤n,m≤100000)
接下来m行,代表m个操作,格式如题目所示。

输出描述:

对于每个2号操作,输出一个整数,表示查询的结果。
示例1

输入

10 10
1 1 2 0
1 3 4 1
2 1 4
1 5 6 2
2 1 6
1 7 8 1
2 3 8
1 8 10 3
2 1 10
2 3 8

输出

2
3
2
4
3

思路:

此题直观的做法是开一个tree[MAXN*4][61]的数组标记,区间修改线段树。可是这样会超内存。题目给出的桶数范围是60,我们就可以用2进制状态压缩。并且2进制取或非常方便。需要注意1<<60时需要强制类型转换为long long。。。


代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAXN 100005
int n,m;
long long tree[4*MAXN],laz[4*MAXN],que;
void pushdown(int v)
{
    laz[v*2]|=laz[v];
    laz[v*2+1]|=laz[v];
    tree[v*2]|=laz[v*2];
    tree[v*2+1]|=laz[v*2+1];
    laz[v]=0;
}
void updata(int v,int L,int R,int ql,int qr,int q)
{
    if(ql<=L && R<=qr)
    {
        tree[v]|=(long long)1<<q;
        laz[v]|=(long long)1<<q;
        return;
    }
    if(laz[v]) pushdown(v);
    int mid=(L+R)/2;
    if(ql<=mid) updata(v*2,L,mid,ql,qr,q);
    if(qr>mid) updata(v*2+1,mid+1,R,ql,qr,q);
    tree[v]=tree[v*2]|tree[v*2+1];
}
void query(int v,int L,int R,int ql,int qr)
{
    if(ql<=L && R<=qr)
    {
        que|=tree[v];
        return;
    }
    if(laz[v]) pushdown(v);
    int mid=(L+R)/2;
    if(ql<=mid) query(v*2,L,mid,ql,qr);
    if(qr>mid) query(v*2+1,mid+1,R,ql,qr);
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(tree,0,sizeof tree);
        memset(laz,0,sizeof laz);
        for(int i=0;i<m;i++)
        {
            int op,l,r,d;
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d%d%d",&l,&r,&d);
                updata(1,1,n,l,r,d);
            }
            else
            {
                scanf("%d%d",&l,&r);
                que=0;
                query(1,1,n,l,r);
                int ans=0;
                while(que)
                {
                    ans+=que&1;
                    que>>=1;
                }
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}
/*
10 10
1 1 2 0
1 3 4 1
2 1 4
1 5 6 2
2 1 6
1 7 8 1
2 3 8
1 8 10 3
2 1 10
2 3 8
*/


这一题还有更简单的做法。用vector[61]存储这个颜色每次放的区间。询问时直接循环遍历61种颜色,看是否存在一个区间与询问区间相交。复杂度为O(61*N);

代码:

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 100005
int n,m;
struct node
{
    int l,r;
}p;
vector<node> vec[61];
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0;i<=60;i++)
            vec[i].clear();
        for(int i=0;i<m;i++)
        {
            int op,l,r,c;
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d%d%d",&l,&r,&c);
                p.l=l;
                p.r=r;
                vec[c].push_back(p);
            }
            else
            {
                scanf("%d%d",&l,&r);
                int ans=0;
                for(int j=0;j<=60;j++)
                {
                    for(int k=0;k<vec[j].size();k++)
                    {
                        if(l<=vec[j][k].r && r>=vec[j][k].l)
                        {
                            ans++;
                            break;
                        }
                    }
                }
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}
/*
10 10
1 1 2 0
1 3 4 1
2 1 4
1 5 6 2
2 1 6
1 7 8 1
2 3 8
1 8 10 3
2 1 10
2 3 8
*/

猜你喜欢

转载自blog.csdn.net/u013852115/article/details/80103824