BZOJ3644 : 陶陶的旅行计划

假设是序列问题,且$S<T$,可以贪心求解,通过维护下述信息进行区间合并。

对于区间$[l,r]$,维护的信息有:

$v$:跳到了$\geq r$的位置后,可以花费$1$往右最多扩展多少。

$f[i]$:从$\leq l+i$开始跳到$\geq r$的位置的最少步数。

$g[i]$:从$\leq l+i$开始跳到$\geq r$的位置时,在$f$最小的基础上,可以花费$0$往右最多扩展多少。

然后树链剖分+线段树维护信息即可。

时间复杂度$O(am\log^2n)$。

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100005,M=262150,K=20;
int n,m,i,x,y,a[N],g[N],v[N<<1],nxt[N<<1],ed,f[N],d[N],size[N],son[N],top[N],loc[N],q[N],dfn;char op;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline void merge(int&f,int&g,int x,int y){if(x<f||x==f&&y>g)f=x,g=y;}
struct P{
int f[K],g[K],v,len;
P(){}
P operator+(const P&b){
P c;
for(int i=0;i<K;i++){
if(i<len){
if(g[i]>=b.len){
c.f[i]=f[i];
c.g[i]=g[i]-b.len;
}else if(g[i]){
c.f[i]=f[i]+b.f[g[i]-1];
c.g[i]=b.g[g[i]-1];
}else c.f[i]=N;
if(v>=b.len)merge(c.f[i],c.g[i],f[i]+1,v-b.len);
else merge(c.f[i],c.g[i],f[i]+b.f[v-1]+1,b.g[v-1]);
}else{
if(i-len>=b.len)break;
c.f[i]=b.f[i-len];
c.g[i]=b.g[i-len];
}
}
c.v=max(v-b.len,b.v);
c.len=len+b.len;
for(int i=1;i<K;i++)if(c.f[i]>c.f[i-1]||c.f[i]==c.f[i-1]&&c.g[i]<c.g[i-1])c.f[i]=c.f[i-1],c.g[i]=c.g[i-1];
for(int i=c.len;i<K;i++)c.f[i]=N,c.g[i]=0;
return c;
}
void set(int x){
v=x,len=1;
f[0]=g[0]=0;
for(int i=1;i<K;i++)f[i]=N,g[i]=0;
}
}vl[M],vr[M],tmp;bool flag;
inline void up(int x){
vl[x]=vl[x<<1]+vl[x<<1|1];
vr[x]=vr[x<<1|1]+vr[x<<1];
}
void build(int x,int a,int b){
if(a==b){
vl[x].set(q[a]);
vr[x].set(q[a]);
return;
}
int mid=(a+b)>>1;
build(x<<1,a,mid),build(x<<1|1,mid+1,b),up(x);
}
void change(int x,int a,int b,int c,int d){
if(a==b){
vl[x].set(d);
vr[x].set(d);
return;
}
int mid=(a+b)>>1;
if(c<=mid)change(x<<1,a,mid,c,d);
else change(x<<1|1,mid+1,b,c,d);
up(x);
}
void askl(int x,int a,int b,int c,int d){
if(c<=a&&b<=d){
if(flag)tmp=tmp+vl[x];else tmp=vl[x],flag=1;
return;
}
int mid=(a+b)>>1;
if(c<=mid)askl(x<<1,a,mid,c,d);
if(d>mid)askl(x<<1|1,mid+1,b,c,d);
}
void askr(int x,int a,int b,int c,int d){
if(c<=a&&b<=d){
if(flag)tmp=tmp+vr[x];else tmp=vr[x],flag=1;
return;
}
int mid=(a+b)>>1;
if(d>mid)askr(x<<1|1,mid+1,b,c,d);
if(c<=mid)askr(x<<1,a,mid,c,d);
}
inline void add(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}
void dfs(int x){
size[x]=1;d[x]=d[f[x]]+1;
for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){
f[v[i]]=x,dfs(v[i]);
size[x]+=size[v[i]];
if(size[v[i]]>size[son[x]])son[x]=v[i];
}
}
void dfs2(int x,int y){
q[loc[x]=++dfn]=a[x];top[x]=y;
if(son[x])dfs2(son[x],y);
for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]&&v[i]!=son[x])dfs2(v[i],v[i]);
}
inline int lca(int x,int y){
for(;top[x]!=top[y];x=f[top[x]])if(d[top[x]]<d[top[y]])swap(x,y);
return d[x]<d[y]?x:y;
}
int cnt,pool[1000][2];
inline int query(int x,int y){
int z=lca(x,y);
flag=cnt=0;
while(top[x]!=top[z]){
askr(1,1,n,loc[top[x]],loc[x]);
x=f[top[x]];
}
askr(1,1,n,loc[z],loc[x]);
while(top[y]!=top[z]){
pool[++cnt][0]=loc[top[y]];
pool[cnt][1]=loc[y];
y=f[top[y]];
}
if(y!=z){
pool[++cnt][0]=loc[z]+1;
pool[cnt][1]=loc[y];
}
for(int i=cnt;i;i--)askl(1,1,n,pool[i][0],pool[i][1]);
return tmp.f[0];
}
int main(){
read(n);
for(i=1;i<=n;i++)read(a[i]);
for(i=1;i<n;i++)read(x),read(y),add(x,y),add(y,x);
dfs(1);
dfs2(1,1);
build(1,1,n);
read(m);
while(m--){
while((op=getchar())!='Q'&&op!='C');
read(x),read(y);
if(op=='C')change(1,1,n,loc[x],y);
else printf("%d\n",query(x,y));
}
return 0;
}

  

上一篇:安卓初級教程(2):SD創建file,儲存與讀寫的方法(1)


下一篇:Linux设备驱动编程之复杂设备驱动