长沙学院2022暑假训练赛（一) -H

作为礼物的字符串

写在前面：

个人感觉这个题目很不错，对个人而言是没见过的并查集优化类型。

思路：

对每个区间对称位置并查集合并是很好想到的一个思路，但时间复杂度显然不对。

直接上题解了。。
先复制一个反串在原串后面方便合并。（类似扩展域的思想）
对单个区间操作时按区间长度划分成$log$个区间和反串合并。
从高次幂向下每次拆分成两个子区间合并。
最后直接查询对称位置即可。

暴力的做法存在很多的重复合并，这么做减小了次数。

时间复杂度： $O(mlo{g}^{2}n)$

具体看代码。

Code:

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N=1e5+20;
int n,m;
int flag[N*200];
// 编号
inline int id(int x,int y){  // x 起点 y 长度幂次
	return x*100+y; // y<=20 乘100即可
}
// 找根
int find(int x){
	if(flag[x]==x) return x;
	else return flag[x]=find(flag[x]);
}
// 合并
void work(int x,int y){
	int s1=find(x),s2=find(y);
	flag[s2]=s1;
}

int main(){
guo312;
	cin>>n>>m;
	// 初始化并查集
	for(int i=1;i<=n*2;i++){
		for(int j=0;j<=20;j++){
			flag[id(i,j)]=id(i,j);  
		}
	}
	// 原位置和反串位置合并
	for(int i=1;i<=n;i++){
		work(id(i,0),id(2*n-i+1,0));
	}
	for(int i=1;i<=m;i++){
		int l1,r; cin>>l1>>r; int len=r-l1+1;
		int l2=2*n-r+1;  // l1 原串起点 l2 反串起点
		for(int j=20;j>=0;j--){
			if(len>>j&1){
				work(id(l1,j),id(l2,j));
				l1+=(1<<j);
				l2+=(1<<j);
			}
		}
	}
	for(int j=20;j>0;j--){  // 高幂次到低幂次
		for(int i=1;i<=n*2;i++){
			int fa=find(id(i,j)); fa/=100;  // 除100得到区间起点
			if(fa==i) continue ;
			// 两个子区间和反串合并
			work(id(i,j-1),id(fa,j-1));  
			work(id(i+(1<<(j-1)),j-1),id(fa+(1<<(j-1)),j-1));
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		if(find(id(i,0))==find(id(n-i+1,0))) ans++;
	}
	if(ans==n) cout<<"YES"<<endl;
	else cout<<"NO"<<endl;
	cout<<ans;
	return 0;
}