hdu 3854 Glorious Array(线段树or树状数组)

题意:n个位置,每个位置有个价值和颜色,颜色为0或1,定义两个位置的距离为两个点之间的价值的最小值,有两个操作,一个是将位置p的颜色取反,另一个是查询当前序列有多少对不同颜色的位置之间的距离小于K。

思路:首先预处理一下,求出某个位置左边第一个比K小的数的位置lf[i]和右边第一个比K小的数的位置rn[i],那么最初的答案就可以这样算:枚举每个位置,计算以该位置为右端点的合法点对的个数,这个只需要计算[1,lf[i]]有多少个和这个点颜色不同的点就好了,因为这些点一定经过lf[i]。对于修改,也利用了类似的思想,当修改一个位置的颜色时,假如从0改到1,那么你就要将[1,lf[i]]和这个颜色不同的减去,相同的加上,右边也一样,注意一下lf[i]==i的情况就好了。求区间里有多少个0 or 1可以使用线段树或树状数组。


代码:


#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-6
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
int sum[maxn<<2],lf[maxn],rn[maxn];
int val[maxn],color[maxn],lz[maxn];
inline void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        if(color[l]==0) sum[rt]=1;
        else sum[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    PushUp(rt);
}
void Update(int p,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]^=1;
        return ;
    }
    int m=(l+r)>>1;
    if(m>=p) Update(p,l,m,rt<<1);
    else Update(p,m+1,r,rt<<1|1);
    PushUp(rt);
}
int Query(int L,int R,int l,int r,int rt)
{
    if(L>R) return 0;
    if(l>=L&&r<=R)
    {
        return sum[rt];
    }
    int m=(l+r)>>1;
    int res=0;
    if(m>=L) res+=Query(L,R,l,m,rt<<1);
    if(m=1;--i)
            if(val[i]


你可能感兴趣的:(数据结构,线段树)