Light bulbs （树状数组模板题）

最新推荐文章于 2024-10-01 05:04:32 发布

weixin_30905133

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量186

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/lengsong/p/11523654.html

版权

There are

A FLIP operation switches the state of a contiguous subset of bulbs.

Given the value of

InputFile

The first line of the input gives the number of test cases,

OutputFile

For each test case, output one line containing Case #x: y, where

样例输入复制

样例输出复制

Case #1: 4
Case #2: 3
题意：n个灯，m次操作，每次把一个区间的灯反转，最初都是暗的，问最后有多少灯是亮的；
题解：树状数组，每次调用add（）函数把L处加一，R+1处减一，然后最后把每一个区间用排序去重之后计算当前区间处L（只用L就够了）被加了多少次（通过调用ask（）函数计算），偶数就代表当前区间的灯都是关的，反之就是开的，然后这个区间长度就是
（str[i+1]-1）-str[i]+1；
还有就是记得内存限制；
最后每一次树状数组清零时是通过保存输入进行清零的，L处减一，R+1处加一；
总的复杂度是O（t*m*log（n））；

#include <bits/stdc++.h>
#pragma GCC optimize(3)
using namespace std;
const int maxn=2e3+5;
int str[maxn],str2[maxn],c[1000005],n,m;
int ask(int x)
{
    int ans=0;
    for(; x; x-=x&-x)ans+=c[x];
    return ans;
}
void add(int x,int y)
{
    for(; x<=n; x+=x&-x)c[x]+=y;
}
template<class T>
inline void read(T&x)
{
    T ans=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        ans=ans*10+ch-'0';
        ch=getchar();
    }
    x=ans*f;
}
int main()
{
    int t,l,r,cnt=0,k=0;
    read(t);
    while(t--)
    {
        cnt=0;
        read(n);
        read(m);
        int m2=m;
        while(m2--)
        {
            read(l);read(r);
            ++l,++r;
            add(l,1);
            add(r+1,-1);
            str[++cnt]=l,str2[cnt]=l;
            str[++cnt]=r+1,str2[cnt]=r+1;
        }
        sort(str+1,str+cnt+1);
        int cnt2=0;
        for(int i=1; i<=cnt; ++i)
        {
            if(str[i]!=str[cnt2])
                str[++cnt2]=str[i];
        }
        int ans=0,flag;
        for(int i=1; i<cnt2; ++i)
        {
            flag=ask(str[i]);
            if((flag&1)==1)
                ans+=str[i+1]-1-str[i]+1;
        }
        printf("Case #%d: %d\n",++k,ans);
        for(int i=1; i<=2*m; i+=2)
        {
            add(str2[i],-1);
            add(str2[i+1],1);
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/lengsong/p/11523654.html