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

行星序列(AHOI2009)(线段树)

时间:2015-05-14 16:42:10      阅读:162      评论:0      收藏:0      [点我收藏+]

标签:

题目描述 Description
“神州“载人飞船的发射成功让小可可非常激动,他立志长大后要成为一名宇航员假期一始,他就报名参加了“小小宇航员夏令营”,在这里小可可不仅学到了丰富的宇航知识,还参与解决了一些模拟飞行中发现的问题,今天指导老师交给他一个任务,在这次模拟飞行的路线上有N个行星,暂且称它们为一个行星序列,并将他们从1至n标号,在宇宙未知力量的作用下这N个行星的质量是不断变化的,所以他们对飞船产生的引力也会不断变化,小可可的任务就是在飞行途中计算这个行星序列中某段行星的质量和,以便能及时修正飞船的飞行线路,最终到达目的地,行星序列质量变化有两种形式:
1,行星序列中某一段行星的质量全部乘以一个值
2,行星序列中某一段行星的质量全部加上一个值
由于行星的质量和很大,所以求出某段行星的质量和后只要输出这个值模P的结果即可,小可可被这个任务难住了,聪明的你能够帮他完成这个任务吗?

输入描述 Input Description
第一行两个整数N和P(1<=p<=1000000000);
第二行含有N个非负整数,从左到右依次为a1,a2,…………,an(0<=ai<=100000000,1<=i<=n),其中ai表示第i个行星的质量:
第三行有一个整数m,表示模拟行星质量变化以及求质量和等操作的总次数。从第四行开始每行描述一个操作,输入的操作有以下三种形式:
操作1:1 t g c 表示把所有满足t<=i<=g的行星质量ai改为ai*c
操作2:2 t g c 表示把所有满足t<=i<=g的行星质量ai改为ai+c
操作3:3 t g 表示输出所有满足t<=i<=g的ai的和模p的值
其中:1<=t<=g<=N,0<=c<=10000000
注:同一行相邻的两数之间用一个空格隔开,每行开头和末尾没有多余空格
输出描述 Output Description
对每个操作3,按照它在输入中出现的顺序,依次一行输出一个整数表示所求行星质量和
样例输入 Sample Input
7 43
1 2 3 4 5 6 7
5
1 2 5 5
3 2 4
2 3 7 9
3 1 3
3 4 7
样例输出 Sample Output
2
35
8
数据范围及提示 Data Size & Hint
100%的数据中,M,N<=100000
40%的数据中,M,N<=10000
题解:这个题和裸线段树唯一不同的地方在于有两种修改,其实这个也很好处理,遇到乘法就直接乘,加法的话先乘后加,注意中间的一些longlong就好了。。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
long long p,t[1000001]={0},a[1000001]={0},lazy[1000001][5]={0}; 
void build(int k,int l,int r)
{
    int mid;
    if (l==r){t[k]=a[l]%p;return;};
    mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    t[k]=(t[k*2]+t[k*2+1])%p;
}
void paint(int k,int l,int r,long long m,long long c)
{
    t[k]=((t[k]*m)%p+(r-l+1)*c)%p;
    lazy[k][1]=(lazy[k][1]*m)%p;
    lazy[k][2]=((lazy[k][2]*m)%p+c)%p;
}
void pushdown(int k,int l,int r)
{
   int mid;
   mid=(l+r)/2;
   paint(k*2,l,mid,lazy[k][1],lazy[k][2]);
   paint(k*2+1,mid+1,r,lazy[k][1],lazy[k][2]);
   lazy[k][1]=1;lazy[k][2]=0;
}
void insert(int k,int l,int r,int ll,int rr,long long m,long long c)
{
    int mid;
    if (l>=ll&&r<=rr){paint(k,l,r,m,c);return;}
    mid=(l+r)/2;
    pushdown(k,l,r);
    if (ll<=mid) insert(k*2,l,mid,ll,rr,m,c);
    if (rr>mid) insert(k*2+1,mid+1,r,ll,rr,m,c);
    t[k]=(t[k*2]+t[k*2+1])%p;    
}
long long qsum(int k,int l,int r,int ll,int rr)
{
   int mid;
   long long ans(0);
   if (l>=ll&&r<=rr) 
     {
       t[k]%=p;
       return t[k];
     }
   mid=(l+r)/2;
   pushdown(k,l,r);
   if (ll<=mid) ans=(ans+qsum(k*2,l,mid,ll,rr))%p;
   if (rr>mid) ans=(ans+qsum(k*2+1,mid+1,r,ll,rr))%p;
   return ans;
}
int main()
{
   int n,m,c,x,y,z,kind,t,g;
   cin>>n>>p;
   long long ans(0);
   for (int i=1;i<=n;i++) scanf("%lld",&a[i]);
   build(1,1,n);
   for (int i=1;i<=500000;i++) lazy[i][1]=1;
   cin>>m;
   for (int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&kind,&t,&g);
        if (kind==1)
        {
            scanf("%lld",&c);
            insert(1,1,n,t,g,c,0);
        }
        if (kind==2)
        {
            scanf("%lld",&c);
            insert(1,1,n,t,g,1,c);
        }
        if (kind==3)
        {
            ans=qsum(1,1,n,t,g)%p;
            printf("%lld\n",ans);
        }
    }
} 

行星序列(AHOI2009)(线段树)

标签:

原文地址:http://blog.csdn.net/sunshinezff/article/details/45720367

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