洛谷P3372/poj3468(线段树lazy_tag)(询问区间和,支持区间修改)
2024-09-01 22:42:28
//线段树 询问区间和,支持区间修改
#include <cstdio> using namespace std; struct treetype
{
int l,r;
long long sum,d;
}; const int maxn=; treetype a[maxn<<];
long long num[maxn]; void build(int k,int l,int r)
{
a[k].l=l;a[k].r=r;a[k].d=;
if (a[k].l==a[k].r)
{
a[k].sum=num[l];
return;
}
int mid=(l+r)>>,i=k<<;
build(i,l,mid);
build(i+,mid+,r);
a[k].sum=a[i].sum+a[i+].sum;
}
void pushdown(int k)
{
if (a[k].l==a[k].r) a[k].d=;
if (a[k].d==) return;
//int i=k<<1;long long t=a[i].r-a[i].l+1;a[i].sum+=t*a[k].d;a[i].d+=a[k].d;
//i++;t=a[i].r-a[i].l+1;a[i].sum+=t*a[k].d;a[i].d+=a[k].d;
int i=k<<;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
i++;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
a[k].d=;
}
void change(int k,int l,int r,long long d)
{
pushdown(k);
if (l<=a[k].l && a[k].r<=r)
{
//long long t=a[k].r-a[k].l+1;
a[k].sum+=(a[k].r-a[k].l+)*d;a[k].d=d;
return;
}
int mid=(a[k].l+a[k].r)>>,i=k<<;
if (l<=mid) change(i,l,r,d);
if (mid<r) change(i+,l,r,d);
a[k].sum=a[i].sum+a[i+].sum;
}
long long query(int k,int l,int r)
{
pushdown(k);
if (l<=a[k].l && a[k].r<=r) return a[k].sum;
int mid=(a[k].l+a[k].r)>>,i=k<<;
long long ans=;
if (l<=mid) ans=query(i,l,r);
if (mid<r) ans+=query(i+,l,r);
return ans;
} int main()
{
int n,m;
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++) scanf("%lld",&num[i]);
build(,,n);
for (int i=;i<=m;i++)
{
int p,x,y;
scanf("%d%d%d",&p,&x,&y);
if (p==)
{
long long k;
scanf("%lld",&k);
change(,x,y,k);
}
else printf("%lld\n",query(,x,y));
}
return ;
}
//线段树 询问区间和,支持区间修改
#include <cstdio> using namespace std; struct treetype
{
int l,r;
long long sum,d;
}; const int maxn=; treetype a[maxn<<];
long long num[maxn]; void build(int k,int l,int r)
{
a[k].l=l;a[k].r=r;a[k].d=;
if (a[k].l==a[k].r)
{
a[k].sum=num[l];
return;
}
int mid=(l+r)>>,i=k<<;
build(i,l,mid);
build(i+,mid+,r);
a[k].sum=a[i].sum+a[i+].sum;
}
void pushdown(int k)
{
if (a[k].l==a[k].r) a[k].d=;
if (a[k].d==) return;
int i=k<<;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
i++;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
a[k].d=;
}
void change(int k,int l,int r,long long d)
{
pushdown(k);
if (l<=a[k].l && a[k].r<=r)
{
a[k].sum+=(a[k].r-a[k].l+)*d;a[k].d=d;
return;
}
int mid=(a[k].l+a[k].r)>>,i=k<<;
if (l<=mid) change(i,l,r,d);
if (mid<r) change(i+,l,r,d);
a[k].sum=a[i].sum+a[i+].sum;
}
long long query(int k,int l,int r)
{
pushdown(k);
if (l<=a[k].l && a[k].r<=r) return a[k].sum;
int mid=(a[k].l+a[k].r)>>,i=k<<;
long long ans=;
if (l<=mid) ans=query(i,l,r);
if (mid<r) ans+=query(i+,l,r);
return ans;
} int main()
{
int n,m;
scanf("%d%d",&n,&m);
for (int i=;i<=n;i++) scanf("%lld",&num[i]);
build(,,n);
for (int i=;i<=m;i++)
{
char p;scanf("%c",&p);
int x,y;
scanf("%c%d%d",&p,&x,&y);
if (p=='C')
{
long long k;
scanf("%lld",&k);
change(,x,y,k);
}
else printf("%lld\n",query(,x,y));
}
return ;
}
最新文章
- [分享] 《步步为营封 Win7》--skyfree
- 跨站脚本 XSS<;一:防御方法>;
- Java Debugging
- Ambari源代码分析之总览
- nginx配置负载均衡与反向代理
- 转载:CEO应向软件工程师学习的7个技能
- InkPicture 控件使用_01
- 李洪强iOS开发Swift篇—05_元组类型
- winfrom存储txt日志函数
- PL/SQL设置关键字大写
- paip.c++ qt 项目工程互相引用的方法
- cf-A. Wet Shark and Odd and Even(水)
- Oracle的海量存储技术
- Kettle的应用——对mysql数据进行表输入与导出
- 【锋利的Jquery】读书笔记三
- SQL SERVER的统计信息
- Python Web 框架:Tornado
- Kibana登录认证设置
- 【Coursera】Technology :Fifth Week(2)
- show tables from information_schema/performance_schema/sys;