【CF734F】Anton and School（构造）

题面

Codeforces
洛谷

题解

算是一道\(easy\)?
发现\((a\&b)+(a|b)=a+b\)。
那么根据给定条件我们就能确定唯一的\(a\)数列，最后再带回去\(check\)一下就做完了？？？

#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long
#define MAX 200200
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int n,a[MAX],b[MAX],c[MAX],s[MAX],cnt[30];ll sum;
bool check()
{
	for(int i=1;i<=n;++i)
	{
		ll sb=0,sc=0;
		for(int j=0;j<30;++j)
			if(a[i]&(1<<j))sc+=(1ll<<j)*n,sb+=(1ll<<j)*cnt[j];
			else sc+=(1ll<<j)*cnt[j];
		if(sb!=b[i]||sc!=c[i])return false;
	}
	return true;
}
int main()
{
	n=read();
	for(int i=1;i<=n;++i)b[i]=read();
	for(int i=1;i<=n;++i)c[i]=read();
	for(int i=1;i<=n;++i)s[i]=b[i]+c[i],sum+=s[i];
	sum/=n+n;for(int i=1;i<=n;++i)a[i]=(s[i]-sum)/n;
	for(int i=1;i<=n;++i)for(int j=0;j<30;++j)cnt[j]+=(a[i]>>j)&1;
	if(!check())printf("-1");
	else for(int i=1;i<=n;++i)printf("%d ",a[i]);
	puts("");return 0;
}