关于卡常
  • 板块题目总版
  • 楼主AKPC
  • 当前回复19
  • 已保存回复20
  • 发布时间2024/5/4 14:32
  • 上次更新2024/5/4 16:25:06
查看原帖
关于卡常
540363
AKPC楼主2024/5/4 14:32

题目 P3313,同题在loj加上Ofast和火车头能过,现在洛谷的最快记录是 2.28s,真的不会卡了,谁能帮个忙吗,感谢!

#include <iostream>
#include <vector>
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)>(b)?(a):(b))
#define swap(a,b) ((a)^=(b)^=(a)^=(b))
char *p1, *p2, buf[55];
#define getchar() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 55, stdin), p1 == p2) ? EOF : *p1 ++ )
int read(){
	int res=0,f=1;char c=getchar();
	while (!isdigit(c)) {if (c=='-') f=-1;c=getchar();}
	while (isdigit(c)) {res=(res<<1)+(res<<3)+(c^48);c=getchar();}
	return res*f;
}
void write(int x){
	if (x<0) putchar('-'),x=-x;
	if (x>9) write(x/10);
	putchar(x%10+48);
}
const int N=2e5+5,SN=605;
int n,q,dep[N],siz[N],fa[N],son[N];
std::vector<int>e[N];
void dfs(int u,int father){
	dep[u]=dep[father]+1;
	fa[u]=father;
	siz[u]=1;
	int maxn=-1;
	for (auto v:e[u]){
		if (v==father) continue;
		dfs(v,u),siz[u]+=siz[v];
		if (siz[v]>maxn) maxn=siz[v],son[u]=v;
	}
}
int dfn[N],top[N],tip=1;
void dfs2(int u,int father,int op){
	if (op) top[u]=top[father];
	dfn[u]=tip++;
	if (son[u]) dfs2(son[u],u,1);
	for (auto v:e[u]){
		if (v==father||v==son[u]) continue;
		dfs2(v,u,0);
	}
}
void init(int root){
	dfs(root,0);
	for (register int i=1;i<=n;i++) top[i]=i;
	dfs2(root,0,0);
}
int _c[N],_w[N],c[N],w[N];
struct block_array{
	#define lc(x) ((x)*val-val+1)
	#define rc(x) (min(n,(x)*val))
	int val,block[N],sum[N][SN],mx[N][SN];
	void build(){
		val=141;
		for (register int i=1;i<=n;i++){
			block[i]=(n-1)/val+1,mx[c[i]][block[i]]=-1e9;
			sum[c[i]][block[i]]+=w[i];
			mx[c[i]][block[i]]=max(mx[c[i]][block[i]],w[i]);
		}
	}
	void cc(int x,int k){
		sum[c[x]][block[x]]-=w[x];
		sum[k][block[x]]+=w[x];
		mx[c[x]][block[x]]=-1e9;
		for (register int i=lc(block[x]);i<=rc(block[x]);i++)
			if (i!=x) mx[c[i]][block[i]]=max(mx[c[i]][block[i]],w[i]);
		mx[k][block[x]]=max(mx[k][block[x]],w[x]);
		c[x]=k;
	}
	void cw(int x,int k){
		sum[c[x]][block[x]]+=k-w[x];
		w[x]=k;
		mx[c[x]][block[x]]=-1e9;
		for (register int i=lc(block[x]);i<=rc(block[x]);i++)
			mx[c[i]][block[i]]=max(mx[c[i]][block[i]],w[i]);
	}
	int qs(int l,int r,int k){
//		cout<<l<<' '<<r<<' '<<k<<'\n';
		int res=0;
		if (block[l]==block[r]){
			for (register int i=l;i<=r;i++) res+=(c[i]==k)*w[i];
			return res;
		}
		for (register int i=l;i<=rc(block[l]);i++) res+=(c[i]==k)*w[i];
		for (register int i=r;i>=lc(block[r]);i--) res+=(c[i]==k)*w[i];
		for (register int i=block[l]+1;i<block[r];i++) res+=sum[k][i];
		return res;
	}
	int qm(int l,int r,int k){
		int res=-1e9;
		if (block[l]==block[r]){
			for (register int i=l;i<=r;i++) if (c[i]==k) res=max(res,w[i]);
			return res;
		}
		for (register int i=l;i<=rc(block[l]);i++) if (c[i]==k) res=max(res,w[i]);
		for (register int i=r;i>=lc(block[r]);i--) if (c[i]==k) res=max(res,w[i]);
		for (register int i=block[l]+1;i<block[r];i++) res=max(res,mx[k][i]);
		return res;
	}
	#undef lc
	#undef rc
}book;
int QS(int x,int y,int k){
	int res=0;
	while (top[x]!=top[y]){
		if (dep[top[x]]<dep[top[y]]) swap(x,y);
		res+=book.qs(dfn[top[x]],dfn[x],k);
		x=fa[top[x]];
	}
	if (dep[x]>dep[y]) swap(x,y);
	res+=book.qs(dfn[x],dfn[y],k);
	return res;
}
int QM(int x,int y,int k){
	int res=-1e9;
	while (top[x]!=top[y]){
		if (dep[top[x]]<dep[top[y]]) swap(x,y);
		res=max(res,book.qm(dfn[top[x]],dfn[x],k));
		x=fa[top[x]];
	}
	if (dep[x]>dep[y]) swap(x,y);
	res=max(res,book.qm(dfn[x],dfn[y],k));
	return res;
}
signed main(){
//	freopen("1.in","r",stdin);
//	freopen("1.out","w",stdout);
	n=read(),q=read();
	for (register int i=1;i<=n;i++){
		_w[i]=read();
		_c[i]=read();
	}
	for (register int i=1;i<n;i++){
		int u=read(),v=read();
		e[u].push_back(v);
		e[v].push_back(u);
	}
	init(1);
	for (register int i=1;i<=n;i++){
		c[dfn[i]]=_c[i];
		w[dfn[i]]=_w[i];
	}
//	for (int i=1;i<=n;i++) cout<<c[i]<<' ';cout<<'\n';
//	for (int i=1;i<=n;i++) cout<<w[i]<<' ';cout<<'\n';
	book.build();
	while (q--){
		char c1=getchar(),c2;
		while (c1!='C'&&c1!='Q') c1=getchar();c2=getchar();
		while (c2!='C'&&c2!='W'&&c2!='S'&&c2!='M') c2=getchar();
		int x=read(),y=read();
		if (c1=='C'&&c2=='C') _c[x]=y,book.cc(dfn[x],y);
		if (c1=='C'&&c2=='W') _w[x]=y,book.cw(dfn[x],y);
		if (c1=='Q'&&c2=='S') printf("%d\n",QS(x,y,_c[x]));
		if (c1=='Q'&&c2=='M') printf("%d\n",QM(x,y,_c[x]));
	}
	return 0;
}
2024/5/4 14:32
加载中...