【算法】树链剖分+线段树(区间加值,区间求和)

【题解】http://hzwer.com/3891.html

中间不要取模不然相减会出错。

血的教训:线段树修改时标记下传+上传,查询时下传。如果修改时标记不下传,下面的结果就会覆盖上面的标记上传造成的影响。

读入后全部排序(离线处理)

链剖之后按顺序每个solve_insert(1,j),对于每次的z询问solve_sum(1,z)。

LCA其实就是两点到达根节点的路径的最近交点。

差分思想的运用:将区间差转为r-(l-1)。

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=,MOD=;
int n,first[maxn],tot=,top[maxn],deep[maxn],pos[maxn],q,ansz[maxn],size[maxn],f[maxn],dfsnum=;
long long anss[maxn*];
struct edge{int from,v;}e[maxn*];
struct node{int l,r,delta,sum;}t[maxn*];
struct numbers{int num,ord;bool flag;}num[maxn*];
void insert(int u,int v)
{tot++;e[tot].v=v;e[tot].from=first[u];first[u]=tot;}
bool cmp(numbers a,numbers b)
{return a.num<b.num;}
void dfs1(int x,int fa)
{
size[x]=;
for(int i=first[x];i;i=e[i].from)
if(e[i].v!=fa)
{
int y=e[i].v;
f[y]=x;
deep[y]=deep[x]+;
dfs1(y,x);
size[x]+=size[y];
}
}
void dfs2(int x,int tp,int fa)
{
pos[x]=++dfsnum;
top[x]=tp;
int k=;
for(int i=first[x];i;i=e[i].from)
if(e[i].v!=fa&&size[e[i].v]>size[k])k=e[i].v;
if(k==)return;
dfs2(k,tp,x);
for(int i=first[x];i;i=e[i].from)
if(e[i].v!=fa&&e[i].v!=k)dfs2(e[i].v,e[i].v,x); }
void build(int k,int l,int r)
{
t[k].l=l;t[k].r=r;t[k].delta=;t[k].sum=;
if(l==r)return;
int mid=(l+r)>>;
build(k<<,l,mid);
build(k<<|,mid+,r);
t[k].sum=t[k<<].sum+t[k<<|].sum;
}
void add(int k,int l,int r)
{
int left=t[k].l,right=t[k].r;
if(l<=left&&right<=r)
{
t[k].delta++;
t[k].sum+=right-left+;
return;
}
if(t[k].delta)
{
t[k<<].delta+=t[k].delta;
t[k<<].sum+=(t[k<<].r-t[k<<].l+)*t[k].delta;
t[k<<|].delta+=t[k].delta;
t[k<<|].sum+=(t[k<<|].r-t[k<<|].l+)*t[k].delta;
t[k].delta=;
}
int mid=(left+right)>>;
if(l<=mid)add(k<<,l,r);
if(r>mid)add(k<<|,l,r);
t[k].sum=t[k<<].sum+t[k<<|].sum;
}
long long query(int k,int l,int r)
{
int left=t[k].l,right=t[k].r;
if(l<=left&&right<=r)return t[k].sum;
if(t[k].delta)
{
t[k<<].delta+=t[k].delta;
t[k<<].sum+=(t[k<<].r-t[k<<].l+)*t[k].delta;
t[k<<|].delta+=t[k].delta;
t[k<<|].sum+=(t[k<<|].r-t[k<<|].l+)*t[k].delta;
t[k].delta=;
}
int mid=(left+right)>>;
long long ans=;
if(l<=mid)ans=query(k<<,l,r);
if(r>mid)ans+=query(k<<|,l,r);
return ans;
}
void solve_ins(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
add(,pos[top[x]],pos[x]);
x=f[top[x]];
}
if(pos[x]>pos[y])swap(x,y);
add(,pos[x],pos[y]);
}
long long solve_sum(int x,int y)
{
long long ans=;
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
ans+=query(,pos[top[x]],pos[x]);
x=f[top[x]];
}
if(pos[x]>pos[y])swap(x,y);
ans+=query(,pos[x],pos[y]);
return ans;
}
int main()
{
scanf("%d%d",&n,&q);
int u;
for(int i=;i<=n;i++)
{
scanf("%d",&u);
insert(u+,i);
insert(i,u+);
}
int ll,rr,zz;
for(int i=;i<=q;i++)
{
scanf("%d%d%d",&ll,&rr,&zz);ll++;rr++;zz++;
ansz[i]=zz;
num[i*-].num=ll-;num[i*-].ord=i;num[i*-].flag=;
num[i*].num=rr;num[i*].ord=i;num[i*].flag=;
}
sort(num+,num+q*+,cmp);
build(,,n);dfs1(,-);dfs2(,,-);
int now=;
memset(anss,,sizeof(anss));
for(int i=;i<=q*;i++)
{
if(num[i].num>now)
for(int j=now+;j<=num[i].num;j++)solve_ins(,j);
now=num[i].num;
if(num[i].flag)anss[num[i].ord]+=solve_sum(,ansz[num[i].ord]);
else anss[num[i].ord]-=solve_sum(,ansz[num[i].ord]);
} for(int i=;i<=q;i++)printf("%lld\n",anss[i]%MOD);
return ;
}

最新文章

  1. Python系统命令操作
  2. JavaScript 运动框架
  3. Best Practices for Performance_1、2 memory、Tips 性能和小的优化点、 onTrimMemory
  4. php 的函数参数值类型限定
  5. ASP.NET定制简单的错误处理页面
  6. 【linux】 linux 查看系统信息
  7. SOA之(4)——服务实现的途径
  8. python20151130
  9. java之表达式陷阱
  10. git从指定的commit创建分支
  11. laravel敏捷应用
  12. 找呀志_通过开源框架引AsyncHttpClient处理get/post要求
  13. 领域模型(Domain Model)
  14. V8 Javascript 引擎设计理念
  15. (十) Jquery的基本使用
  16. Spring MVC 项目搭建 -6- spring security 使用自定义Filter实现验证扩展资源验证,使用数据库进行配置
  17. sql注入--双查询报错注入原理探索
  18. 四、OpenStack—glance组件介绍与安装
  19. 使用log4net生成日志文件
  20. 一些有价值的Blog推荐

热门文章

  1. 解决android invalid symbol: &#39;switch&#39;
  2. iOS-修改导航栏文字字体和颜色
  3. Linux防火墙iptables学习
  4. TCP系列39—拥塞控制—2、拥塞相关算法及基础知识
  5. thinkphp5学习记录一
  6. Python使用ElementTree美化XML格式
  7. java基础--逻辑运算符-- 002
  8. jenkins部署springboot多项目
  9. [C/C++] C++类对象创建问题
  10. html5 isPointInPath相关操作