2021.02.28 一些好题的记录（二）

$$\text{CF1304E 1-Trees and Queries}$$

$\text{[Problem]}$

$\text{[Solution]}$

借鉴 NOIp2019 PJ 的第 $4$ 题的思路，我们可以发现，如果存在一条从 $a$ 到 $b$ 的长度为 $t$ 的路，则只要同时满足：

$$t\le k,t\equiv k(\mathrm{m}\mathrm{o}\mathrm{d}2)$$

答案就是 Yes（因为你可以在任何一条边上走来走去，每次走将会多出 $2$ 的长度，毕竟是一来一回）。

于是我们只需要求出从 $a$ 到 $b$ 的路径长度即可。

路径分成 $3$ 种：

1. 直接从 $a$ 到 $b$（即原树上的边）
2. 从 $a$ 到 $x$ 再到 $y$ 再到 $b$
3. 从 $a$ 到 $y$ 再到 $x$ 再到 $b$

用 LCA 算法求出树上两点间距离，直接判断即可。

$\text{[code]}$

const int N=1e5+100;
struct edge{//链式前向星 
	int next,to;//的模板 
}e[N<<1];int h[N],te;
inline void add(int u,int v){
	e[++te]=(edge){h[u],v};h[u]=te;
}
int dep[N],f[22][N],n,q;
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	for(int i=1;i<=20;i++)
		f[i][u]=f[i-1][f[i-1][u]];
	for(int i=h[u];i;i=e[i].next){
		register int v=e[i].to;
		if (v==fa) continue;//判重 
		f[0][v]=u;dfs(v,u);
	}
}
inline int LCA(int u,int v){
	if (dep[u]<dep[v]) swap(u,v);
	for(int i=20;i>=0;i--)
		if (dep[f[i][u]]>=dep[v])
			u=f[i][u];
	if (u==v) return u;
	for(int i=20;i>=0;i--)
		if (f[i][u]!=f[i][v])
			u=f[i][u],v=f[i][v];
	return f[0][u];
}
inline int dist(int u,int v){
	return dep[u]+dep[v]-2*dep[LCA(u,v)];
}
inline bool check(int t,int k){
	return t<=k&&(k-t)%2==0;
}
int main(){
	n=read();//输入树的大小 
	for(int i=1,u,v;i<n;i++){
		u=read();v=read();
		add(u,v);add(v,u);
	}
	dfs(1,0);q=read();
	for(int i=1;i<=q;i++){
		int x=read(),y=read(),a=read(),b=read(),k=read();
		int ab=dist(a,b),ax=dist(a,x),ay=dist(a,y),bx=dist(b,x),by=dist(b,y);
		if (check(ab,k)||check(ax+by+1,k)||check(ay+bx+1,k)) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}

---

$$\text{CF1322B Present}$$

$\text{[Problem]}$

$\text{[Solution]}$

因为是二进制的操作，所以我们考虑二进制下的答案（下面的讨论是针对二进制的）。

显然每一位是互相独立的，所以我们只考虑第 $k$ 位，最后在统计每一位的答案即为总答案。

记 $b_i=a_i\mathrm{m}\mathrm{o}\mathrm{d}\left(2^{k+1}-1\right)$，即 $a_i$ 的末 $k$ 位值。

如果第 $k$ 位对答案有贡献，则代表有奇数个 $b_i+b_j$ 的第 $k$ 位为 $1$。

现在的问题是，如何求出有多少对 $(i,j)$ 满足 $b_i+b_j$ 的第 $k$ 位为 $1$。

如果 $b_i+b_j\in [2^k,2^{k+1}-1]$，那么 $(i,j)$ 是满足条件的。

但是我们还要考虑这样的情况：第 $k$ 位进位了，然后第 $k-1$ 位进位了，最后第 $k$ 位为 $1$。这个情况有点复杂，但是我们可以证明，此时 $b_i+b_j\in [2^{k+1}+2^k,2^{k+2}-2]$。

现在的问题是，如何快速的求出有多少对 $(i,j)$ 满足 $b_i+b_j\in [L,R]$。

显然，$b$ 数组内部的顺序对答案没有影响，所以我们将 $b$ 数组排序。然后的事情就可以用 Two-pointers 解决了。

$\text{[code]}$

const int N=4e5+100;
int a[N],b[N],n,ans;
inline int count(int u,int v){
	register int sum=0;//统计和 
	for(int i=n,l=1,r=1;i>=1;i--){// *1
		while (l<=n&&b[i]+b[l]<u) l++;
		while (r<=n&&b[i]+b[r]<=v) r++;
		sum+=r-l-(l<=i&&i<r);//*2 
	}
	return (sum>>1)&1;//*3
}
inline int calc(int k){
	for(int i=1;i<=n;i++)
		b[i]=a[i]&((1<<(k+1))-1);
	sort(b+1,b+n+1);//记得排序 
	return count(1<<k,(1<<(k+1))-1)^count(3<<k,(1<<(k+2))-2);
}
inline void calc_answer(){
	for(int i=0;i<27;i++)
		ans+=calc(i)<<i;
}
int main(){
	n=read();ans=0;
	for(int i=1;i<=n;i++)
		a[i]=read();
	calc_answer();
	printf("%d",ans);
	return 0;
}

需要注意的代码细节：

1. 代码 $1$ 处，注意 $i$ 应该是倒序枚举，否则 Two-pointers 是错的。
2. 代码 $2$ 处：注意当 $i\in [l,r]$ 时，$i$ 是不能算入 $\text{sum}$，因为任意 $(i,i)$ 都是不合法的。
3. 代码 $3$ 处：注意上面我们把每个 $(i,j)$ 算了两次（$i$ 一次，$j$ 一次），所以注意先除以二。
4. 代码涉及位运算处：注意位运算的优先级低于加减，所以勤加括号。