当时是在疯狂刷线段树类的题目,刷到这一题时,一点思路都没有,后来看到了大佬的博客,恍然大悟。
大佬链接
原题链接
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r
#define ll long long
using namespace std;
const int MX=200009;
const ll inf=1e18;
int n;
int L[MX],R[MX],C[MX],t[MX<<2];
ll dis[MX];
struct node{
int x;
node(int x=0):x(x){}
bool operator < (const node &a) const {
return C[x]+dis[x]>C[a.x]+dis[a.x]; // 这里是做的真的棒,为了初始化优先队列,使当前优先队列中所储存的路径中,路径已花费+该路径即将花费总和最低的,排第一
}
};
priority_queue<node> que;
void build(int k,int l,int r){
if( l==r ){
t[k]= (l==1 ? 0 : 1 );
return ;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
t[k]=t[k<<1]|t[k<<1|1]; // 向上进行剪枝,因为找的是最短的路径,所以每个节点只需要更新一次,更新完后,关掉该节点的路径,也就是向上剪枝
}
void update(int k,int l,int r,int L,int R,ll w){
if( t[k]==0 ) //已经更新过的就不会再去碰了
return ;
if( l==r ){ // 只要能更新,它必定是最短路径,并且已经储存的路径(dis已经赋过值的路径)中不存在比它更短的
t[k]=0; // 以后不会在去碰走过节点
dis[l]=w;
que.push(node(l)); // 将它存起来,以便剩下的还未更新到的节点继续更新
return ;
}
int mid=(l+r)>>1;
if( L<=mid )
update(lson,L,R,w);
if( mid<R )
update(rson,L,R,w);
t[k]=t[k<<1]|t[k<<1|1]; // 向上剪枝
}
int main()
{
// freopen("input.txt","r",stdin);
int T;
scanf("%d",&T);
while( T-- ){
scanf("%d",&n);
for( int i=1 ; i<=n ; i++ ) scanf("%d",&L[i]);
for( int i=1 ; i<=n ; i++ ) scanf("%d",&R[i]);
for( int i=1 ; i<=n ; i++ ) scanf("%lld",&C[i]);
for( int i=1 ; i<=n ; i++ ) dis[i]=inf;
dis[1]=0; // 将每个地点的路径长初始化为无穷大
build(1,1,n);
que.push(node(1));
while( !que.empty() ){
int u=que.top().x;
que.pop();
int l=max(0,u-R[u]); // 以下都是更新左右两边的目的地,但要注意范围不能超
int r=max(0,u-L[u]);
if( r>=1 )
update(1,1,n,max(1,l),r,C[u]+dis[u]);
l=min(n+1,u+L[u]);
r=min(n+1,u+R[u]);
if( l<=n )
update(1,1,n,l,min(n,r),C[u]+dis[u]);
}
for( int i=1 ; i<=n ; i++ ){
if( dis[i]==inf )
dis[i]=-1;
printf("%lld%c",dis[i],i==n?'\n':' ');
}
}
return 0;
}