HDU-4288:Coder(线段树+离线操作)


题目链接:点击打开链接


题目大意:

给你一些数让你依次添加到一个集合中,集合中的数是递增的。并且中间会求集合中下标mod5=3的数的和


解题思路:

自己完全不会做,表示线段树怎么能用来写这种东西。

后来还是给大神跪了。据说可以暴力压时间水过,不过也没尝试。

具体的思路就是先将所有数离线保存起来,然后依次再加入线段树中,具体操作也说不太清楚。

中间有一步左加右减的操作。很神奇。先将代码搬过来 t[rt].sum[i]=t[lson].sum[i]+t[rson].sum[(i-t[lson].cnt%5+5)%5];  cnt表示当前节点有效数字

可以看到当前节点 mod5=i 的值是由左节点 mod5=i 的值+右节点 mod5=x(暂且用x表示)。x=(i-t[lson].cnt%5+5)%5

这部分可能有些难以理解,举个例子说明吧   假设现在左节点有4个有效数字 ,右节点有4个有效数字。  那么他们在集合中的下标理论上是 1到8.

假设现在求 mod5=3 的值。那么左儿子就要取 下标mod5=3 的那些数,左儿子下标是正常从1开始的。但是右儿子的有效数字虽然集合中的下标是从5到8,但是在它的线段树中实际上还是1到4,那么我们当前节点mod5=3的数应该有两个 下标为3和下标为8。但是右儿子现在下标不是从5开始,所以就需要转化一下即  x=(i-t[lson].cnt%5+5)%5

下面贴代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define rank ra
#define lson rt<<1
#define rson rt<<1|1
#define pb push_back
using namespace std;
typedef long long ll;
int n,m,ans;
int a[100010],v[100010];
char c[100010][5];
struct node
{
    int l,r,mid;
    int cnt;        //记录有效数字的个数
    ll sum[5];      //分别mod5=i的和
}t[500000];
void pushup(int rt)
{
    t[rt].cnt=t[lson].cnt+t[rson].cnt;  //更新有效数字
    for(int i=0;i<5;i++)        //更新当前值得和
        t[rt].sum[i]=t[lson].sum[i]+t[rson].sum[(i-t[lson].cnt%5+5)%5];
}
void build(int l,int r,int rt)  //初始化
{
    int mid=(l+r)>>1;
    t[rt].l=l;t[rt].r=r;
    t[rt].mid=mid;
    for(int i=0;i<5;i++)
        t[rt].sum[i]=0;
    t[rt].cnt=0;
    if(l==r)
        return ;
    build(l,mid,lson);
    build(mid+1,r,rson);
}
void update(int idx,int flag,int rt)
{
    if(t[rt].l==t[rt].r)
    {
        if(flag)        //更新操作
        {
            t[rt].sum[1]=v[idx];
            t[rt].cnt=1;
        }
        if(flag<0)
        {
            t[rt].sum[1]=0;
            t[rt].cnt=0;
        }
        return ;
    }
    if(idx<=t[rt].mid)
        update(idx,flag,lson);
    if(idx>t[rt].mid)
        update(idx,flag,rson);
    pushup(rt);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int p=0;
        for(int i=1;i<=n;i++)
        {
            scanf(" %s",c[i]);
            if(c[i][0]!='s')
            {
                scanf("%d",&a[i]);
                v[++p]=a[i];        //离线储存
            }
        }
        int k=1;
        sort(v+1,v+1+p);        //排序去重
        for(int i=2;i<=p;i++)
        {
            if(v[i]!=v[i-1])
                v[++k]=v[i];
        }
        build(1,k,1);
        for(int i=1;i<=n;i++)
        {
            if(c[i][0]=='s')
                cout<


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