李超线段树 $MLE$
查看原帖
李超线段树 $MLE$
1255395
Chenliy楼主2024/7/13 19:44

MLE on #6

求助万能的谷民

//#define NDEBUG
#include<bits/stdc++.h>
#define For(i,j,k) for(int i=(j);i<=(k);++i)
#define ForD(i,j,k) for(int i=(j);i>=(k);--i)
#define int long long
using namespace std;
bool Mbe;

int tool=0;

// k:b[u] b:f[u]
const int N=1e5+5,B=1e5;
const int inf=1e18;
int head[N],nxt[N<<1],to[N<<1],ecnt=0;
void add(int u,int v){
	nxt[++ecnt]=head[u];
	head[u]=ecnt;
	to[ecnt]=v;
}
#define pb push_back
int cnt=0,tot=0,a[N],rt[N],del[N<<3],mn[N<<3],ls[N<<3],rs[N<<3];
int f[N],b[N];
int node(){
	int x=(tot?del[tot--]:++cnt);
	return x;
}
int Del(int u){
	ls[u]=rs[u]=mn[u]=0;
	del[++tot]=u;
}
int get(int id,int p){
	return f[id]+p*b[id];
}
void add(int &u,int l,int r,int id){
	assert(u<=N);
	if(!u){
		u=node();
		ls[u]=rs[u]=0;
		mn[u]=id;
		return;
	}
	if(get(mn[u],l)>=get(id,l)&&get(mn[u],r)>=get(id,r)) return mn[u]=id,void();
	if(get(mn[u],l)<=get(id,l)&&get(mn[u],r)<=get(id,r)) return;
	int mid=(l+r)/2;
	int &a=id,&b=mn[u];
	if(get(a,mid)<get(b,mid)) swap(a,b);
	if(get(a,l)<get(b,l)) add(ls[u],l,mid,a);
	else add(rs[u],mid+1,r,a);
}
int merge(int u,int v,int l,int r){
	if(!u||!v) return u|v;
	int mid=(l+r)/2;
	ls[u]=merge(ls[u],ls[v],l,mid);
	rs[u]=merge(rs[u],rs[v],mid+1,r);
	add(u,l,r,mn[v]);
	Del(v);
}
int query(int u,int l,int r,int x){
	if(!u) return inf;
	int ans=get(mn[u],x);
	int mid=(l+r)/2;
	if(x<=mid) ans=min(ans,query(ls[u],l,mid,x));
	else ans=min(ans,query(rs[u],mid+1,r,x));
	return ans;
}
void dfs(int u,int fa){
//	cerr<<u<<' '<<cnt<<' '<<tot<<endl;
	bool flg=0;
	for(int i=head[u];i;i=nxt[i]){
		int v=to[i];
		if(v!=fa){
			flg=1;
			dfs(v,u);
			rt[u]=merge(rt[u],rt[v],-B,B);
		}
	}
	f[u]=(flg?query(rt[u],-B,B,a[u]):0);
	add(rt[u],-B,B,u);
}
void solve(){
	int n; cin>>n;
	
	if(n==63054) tool=1;
	
	For(i,1,n) cin>>a[i];
	For(i,1,n) cin>>b[i];
	For(i,1,n-1){
		int u,v; cin>>u>>v;
		add(u,v),add(v,u);
	}
	dfs(1,0);
	For(i,1,n) cout<<f[i]<<' '; cout<<endl;
}
bool Med;
signed main(){
	fprintf(stderr,"%.3lf MB\n",(&Mbe-&Med)/1048576.0);
//	freopen("CF932F.in","r",stdin), freopen("CF932F.out","w",stdout);
	ios::sync_with_stdio(false); cin.tie(0),cout.tie(0);
	int _=1;
//	cin>>_;
	while(_--) solve();
	cerr<<1e3*clock()/CLOCKS_PER_SEC<<" ms\n";
	return 0;
}
2024/7/13 19:44
加载中...