CF1689E ANDfinity 题解

CF1689E ANDfinity

题意

给你一个无向图，节点 $i$ 和 $j$ 有边当且仅当 $a_i\&a_j>0$，有两种操作：

1. 选择一个元素 $a_i$ 并将它加一。
2. 选择一个元素 $a_i$ 并将它减一（仅能在 $a_i>0$ 时做此操作）。

求让这个图连通的最小操作次数。

解析

首先我们发现，若 $a_i=0$，则 $i$ 不可能与任何节点连通，所以先对所有 $a_i=0$ 先加一。

接下来对正整数序列 $a$ 进行考虑。

假设 $\text{lowbit}(a_i)$ 最大的元素组成的集合为 $\text{S}$。

若 $|\text{S}|=1$，将这个元素减一后就会与所有的元素连通（减完一后所有的低位都会变成 $1$），即答案为一。

若 $|\text{S}|>1$，还是将其中一个元素减一，还是一样，这个元素就会与除去 $\text{S}$ 以外的所有元素连通，接下来只需要，将另外一个 $\text{S}$ 中的元素加一，就能保证 $\text{S}$ 中的元素与其他元素联通了，即答案为 $2$。

用并查集维护连通即可，时间复杂度 $O(n^2{\log }_2{a}_i)$。

附代码：

#include<bits/stdc++.h>
#define INF 1e16
#define endl '\n'
#define PI acos(-1)
#define pb push_back
#define int long long
#define mem(x,v) memset(x,v,sizeof x)
using namespace std;
const int N = 2e3+5;
inline int read(){
	register int x=0,f=1;register char c=getchar();
	while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
	return x*f;
}

int lowbit(int x){return x&(-x);}
int T,n,ans,a[N];
struct Set{
	int fa[N],val[N],siz[N];
	void init(int n){for(int i=1;i<=n;i++)fa[i]=i,val[i]=a[i],siz[i]=0;}
	int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
	int merge(int x,int y){
		x=find(x),y=find(y);
		if(x==y)return 0;
		if(siz[x]<siz[y])fa[x]=y,siz[y]=max(siz[y],siz[x]+1),val[y]|=val[x];
		else fa[y]=x,siz[x]=max(siz[x],siz[y]+1),val[x]|=val[y];
		return 1;
	}
	int query(int x){return val[find(x)];}
}S;
int ck(){
	int cnt=0;S.init(n);
	for(int i=1;i<=n;i++)for(int j=1;j<i;j++)if(a[i]&a[j]){
		if(S.merge(i,j))cnt++;
		if(cnt==n-1)return 1;
	}
	return 0;
}
int out(){
	cout<<ans<<endl;
	for(int i=1;i<=n;i++)cout<<a[i]<<" ";
	cout<<endl;
}

void solve(){
	cin>>n,ans=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		if(!a[i])a[i]++,ans++;
	}
	S.init(n);
	if(ck()){out();return;}
	for(int i=1;i<=n;i++){
		a[i]++,ans++;
		if(ck()){out();return;}
		a[i]-=2;
		if(ck()){out();return;}
		a[i]++,ans--;
	}
	int mx=0;
	for(int i=1;i<=n;i++)mx=max(mx,lowbit(a[i]));
	for(int i=1;i<=n;i++){
		if(mx==lowbit(a[i])){
			a[i]--,ans++;
			if(ck()){out();return;}
			break;
		}
	}
	for(int i=1;i<=n;i++)if(!(S.query(i)&1)){a[i]++,ans++,out();return;}
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	for(cin>>T;T--;)solve();
	return 0;
}