【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;

}