码迷,mamicode.com
首页 > 其他好文 > 详细

【BZOJ-4631】踩气球 线段树 + STL

时间:2016-08-21 21:21:42      阅读:545      评论:0      收藏:0      [点我收藏+]

标签:

4631: 踩气球

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 224  Solved: 114
[Submit][Status][Discuss]

Description

六一儿童节到了, SHUXK 被迫陪着M个熊孩子玩一个无聊的游戏:有N个盒子从左到右排成一排,第i个盒子里装着Ai个气球。
SHUXK 要进行Q次操作,每次从某一个盒子里拿出一个没被踩爆的气球,然后熊孩子们就会立刻把它踩爆。
这M个熊孩子每个人都指定了一个盒子区间[Li, Ri]。 如果某一个时刻,一个熊孩子发现自己选定的盒子区间[Li, Ri]中的所有气球都已经被踩爆了,他就会非常高兴(显然之后他一直会很高兴)。
为了不辜负将自己的任务强行塞给 SHUXK 的那个人的期望, SHUXK 想向你询问: 
他每次操作过后会有多少个熊孩子很高兴。

Input

第一行包含两个正整数N和M,分别表示盒子和熊孩子的个数。
第二行包含N个正整数Ai( 1 < = Ai < = 10^5),表示每个盒子里气球的数量。
以下M行每行包含两个正整数Li, Ri( 1 < = Li < = Ri < = N),分别表示每一个熊孩子指定的区间。
以下一行包含一个正整数Q,表示 SHUXK 操作的次数。
以下Q行每行包含一个正整数X,表示这次操作是从第X个盒子里拿气球。为了体现在线,我们对输入的X进行了加密。
假设输入的正整数是x‘,那么真正的X = (x‘ + Lastans − 1)Mod N + 1。其中Lastans为上一次询问的答案。对于第一个询问, Lastans = 0。
输入数据保证1 < = x‘ < = 10^9, 且第X个盒子中有尚未被踩爆的气球。
N < = 10^5 ,M < = 10^5 ?,Q < = 10^5

Output

包含Q行,每行输出一个整数,表示 SHUXK 一次操作后询问的
答案。答案的顺序应与输入数据的顺序保持一致。

Sample Input

5 3
1 1 1 1 1
5 5
2 2
1 3
5
4
2
5
2
3

Sample Output

0
1
1
2
3
【样例说明】
实际上每次操作的盒子是: 4 2 1 3 5
在第二次操作后,第二个熊孩子会高兴 (区间[2,2]中的气球已经全部被踩爆)。
在第四次操作后,第三个熊孩子会高兴(区间[1,3]中的气球已经全部被踩爆)。
在第五次操作后,第一个熊孩子会高兴(区间[5,5]中的气球已经全部被踩爆)。

HINT

Source

Solution

比较好想的一道题

首先对序列建线段树,把M个区间建到线段树上,在线段树的相应节点上记录

维护区间的A[]值和

修改操作相当于单点-1

当一个区间的和=0时更新这个区间上的熊孩子区间的答案,然后统计ans

期望的时间复杂度大概是$O(MlogN)$

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<0 || ch>9) {if (ch==-) f=-1; ch=getchar();}
    while (ch>=0 && ch<=9) {x=x*10+ch-0; ch=getchar();}
    return x*f;
}
#define MAXN 100010
int N,M,Q,size[MAXN],ans,last,A[MAXN];
struct SegmentTreeNode{int l,r,sum; vector<int>v; }tree[MAXN<<2];
inline void Update(int now) {tree[now].sum=tree[now<<1].sum+tree[now<<1|1].sum;}
inline void PushUp(int now)
{
    if (tree[now].sum) return;
    int len=tree[now].v.size(),l=tree[now].l,r=tree[now].r;
    for (int i=0; i<=len-1; i++)
        size[ tree[now].v[i] ]-=r-l+1;
    for (int i=0; i<=len-1; i++)
        if (!size[ tree[now].v[i] ]) ans++;
    tree[now].v.clear();
} 
void BuildTree(int now,int l,int r)
{
    tree[now].l=l; tree[now].r=r;
    if (l==r) {tree[now].sum=A[l]; return;}
    int mid=(l+r)>>1;
    BuildTree(now<<1,l,mid);
    BuildTree(now<<1|1,mid+1,r);
    Update(now); PushUp(now);
}
inline void Change(int now,int pos,int D)
{
    
    int l=tree[now].l,r=tree[now].r;
    if (l==r) {tree[now].sum+=D;  PushUp(now); return;}
    int mid=(l+r)>>1;
    if (pos<=mid) Change(now<<1,pos,D);
    if (pos>mid) Change(now<<1|1,pos,D);
    Update(now); PushUp(now);
}
inline void Cover(int now,int L,int R,int id)
{
    int l=tree[now].l,r=tree[now].r;
    if (L<=l && R>=r) {tree[now].v.push_back(id); size[id]=R-L+1; return;}
    int mid=(l+r)>>1;
    if (L<=mid) Cover(now<<1,L,R,id);
    if (R>mid) Cover(now<<1|1,L,R,id);
    Update(now); PushUp(now);
}
inline int GetX(int x) {return (x+last-1)%N+1;}
int main()
{
    N=read(),M=read();
    for (int i=1; i<=N; i++) A[i]=read();
    BuildTree(1,1,N);
    for (int L,R,i=1; i<=M; i++) L=read(),R=read(),Cover(1,L,R,i);
    Q=read();
    for (int x,i=1; i<=Q; i++) x=GetX(read()),Change(1,x,-1),printf("%d\n",last=ans);
    return 0;
}

总感觉有种不科学的....毕竟就用了10分钟就A了...

【BZOJ-4631】踩气球 线段树 + STL

标签:

原文地址:http://www.cnblogs.com/DaD3zZ-Beyonder/p/5793494.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!