CF1689 E 题解

Sakura_xyz

已于 2022-06-13 22:02:50 修改

阅读量173

点赞数

文章标签：图论 c++

于 2022-06-13 21:59:56 首次发布

本文链接：https://blog.csdn.net/weixin_51060666/article/details/125268321

版权

这篇博客探讨了一个算法问题，涉及数组元素的低二进制位操作和构建最大连通分量。文章首先定义了lowbit(x)为x的最低位1及其后续0，并通过两种情况解释如何通过改变数组元素来确保连通性。第一种情况是数组中只有一个元素的lowbit值最大，第二种情况是存在至少两个这样的元素。文章提供了C++代码实现，并通过检查初始化状态和调整元素值来寻找满足条件的解。最后，给出了求解问题的完整过程及答案。

摘要由CSDN通过智能技术生成

题解

现给出构造：
定义 $l o w b i t (x)$ 为 $x$ 二进制最低位 $1$ 与后面的 $0$ 所组成的数。
首先，把所有的 $0$ 改成 $1$ 。
令 $m x$ 为 $max_{i=1}^{n}{lowbit(a_i)}$ 。
若 $a$ 中仅存在一个数 $x$ 使得 $l o w b i t (x) = m x$ ，则将 $x$ 更改为 $x - 1$ ， $x - 1$ 的二进制最低 $log_2{lowbit(x)}$ 位均为 $1$ ，此时由于其他数的 $l o w b i t$ 取值均小于 $x$ ，因此， $x - 1$ 和其他节点均有连边。
若 $a$ 中存在至少两个数 $x, y$ 使得 $l o w b i t (x), l o w b i t (y) = m x$ ，将 $x$ 变为 $x - 1$ ， $y$ 变为 $y + 1$ ，此时显然 $y + 1$ 与其他 $l o w b i t (t) = m x$ 的 $t$ 有连边， $x - 1$ 与所有 $l o w b i t (s) < m x$ 的 $t$ 有连边， $x - 1$ 与 $y + 1$ 间有连边，结论成立。
证毕。

代码

#include<iostream>
#include<vector>
#define MAXN 3005

using namespace std;

#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")

char buf[1<<21],*p1,*p2,ch;
#define getc() p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;

template <typename T>
inline void read(T & s){
	s=0; char ch=getc(); bool f=false;
	while(!std :: isdigit(ch)) (ch=='-'?f=true:0),ch=getc();
	while(std :: isdigit(ch)) s=(s<<3)+(s<<1)+(ch^48),ch=getc();
	(f?s=-s:0);
}

template <typename T>
inline void write(T x){
	if(x<0) putchar('-'),x=-x;
	if(x>9) write(x/10);
	putchar(x%10+48);
}

int n,a[MAXN];

int fa[MAXN];

int Find(int x) { return x==fa[x]?x:fa[x]=Find(fa[x]); }

void merge(int x,int y){
	x=Find(x); y=Find(y);
	fa[x]=y;
}

bool vis[MAXN];

vector <int> edge[MAXN];

void dfs(int x){
	if(vis[x]) return;
	vis[x]=true;
	for(int i : edge[x]) dfs(i);
}

bool checkinit(){
	for(int i=1;i<=n;i++) vis[i]=false,edge[i].clear(),fa[i]=i;
	for(int i=1;i<=n;i++){
		for(int j=i+1;j<=n;j++){
			if(a[i]&a[j]){
				if(Find(i)==Find(j)) continue;
				merge(i,j);
				edge[i].push_back(j);
				edge[j].push_back(i);
			}
		}
	}
	dfs(1);
	for(int i=1;i<=n;i++) if(!vis[i]) return false;
	return true;
}

int lowbit(int x) { return x&(-x); }

bool check(int x){
	for(int i=1;i<=n;i++) vis[i]=false;
	vis[x]=true;
	for(int i=1;i<=n;i++) if(a[x]&a[i]) dfs(i);
	for(int i=1;i<=n;i++) if(!vis[i]) return false;
	return true;
}

void solve(){
	read(n);
	int ans=0;
	for(int i=1;i<=n;i++) read(a[i]);
	for(int i=1;i<=n;i++) if(!a[i]) ans+=++a[i];
	if(checkinit()){
		write(ans); putchar('\n');
		for(int i=1;i<=n;i++) write(a[i]),putchar(" \n"[i==n]);
		return;
	}
	for(int i=1;i<=n;i++){
		a[i]--;
		if(check(i)){
			write(ans+1); putchar('\n');
			for(int i=1;i<=n;i++) write(a[i]),putchar(" \n"[i==n]);
			return;
		}
		a[i]++;
	}
	for(int i=1;i<=n;i++){
		a[i]++;
		if(check(i)){
			write(ans+1); putchar('\n');
			for(int i=1;i<=n;i++) write(a[i]),putchar(" \n"[i==n]);
			return;
		}
		a[i]--;
	}
	int tmp=0;
	for(int i=1;i<=n;i++) tmp=max(tmp,lowbit(a[i]));
	for(int i=1;i<=n;i++){
		if(lowbit(a[i])==tmp){
			a[i]--;
			break;
		}
	}
	for(int i=n;i>=1;i--){
		if(lowbit(a[i])==tmp){
			a[i]++;
			break;
		}
	}
	write(ans+2); putchar('\n');
	for(int i=1;i<=n;i++) write(a[i]),putchar(" \n"[i==n]);
}

int main(){
	int T; read(T);
	while(T--) solve();
	return 0;
}