牛客小白月赛25-J异或和之和——位运算与组合计数
题目链接: https://ac.nowcoder.com/acm/contest/5600/J
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/8c1ebc75fdaa92607dec7404969945f5.png#pic_center)
题意:
在一个数组中任取三个数的异或和之和。
异或:二进制下相同为0,不同为1。如:
0 ^ 1 = 1, 0 ^ 1 = 1, 1 ^ 1 = 0, 0 ^ 0 = 0。
在任取三个数中,只有两种情况的异或为1:1 1 1 和 0 0 1
所以一开始就要用一个异或数组来记录每个数二进制下的1的个数。
for(int i = 0;i < n; i++)
{
ll x;
cin >> x;
ll k = 0;
while(x) // 二进制下的每一位的1储存下来,N-t[i]就是0的个数
{
if(x&1)
XOR[k]++;
k++;
x >>= 1;
}
}
当在所有1 1 1的情况下有C(XOR[i], 3)种,所有的1里任取3个;在所有0 0 1的情况下有C(n - XOR[i], 2) * XOR[i])种,所有的0里人任取2个在和任意一个1组合。
由于处理组合数时,分子分母很大,需要用到费马小定理
AC code
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
#define INF 0x7f7f7f
#define mem(a,b) memset(a , b , sizeof(a))
#define FOR(i, x, n) for(int i = x;i <= n; i++)
const ll mod = 1e9 + 7;
// const int maxn = 100005;
ll FPM(ll a, ll b)
{
ll ans = 1;
while(b)
{
if(b & 1)
{
ans = ans * a % mod;
}
a = a * a % mod;
b >>= 1;
}
return ans % mod;
}
ll C(ll n, ll m)
{
if(n < m)
return 0;
ll ans = 1;
for(ll i = 1;i <= m; i++)
{
ans = (ans % mod * (n - i + 1) % mod * FPM(i, mod - 2) % mod) % mod;
}
return ans % mod;
}
ll XOR[64];
void solve()
{
ll n;
cin >> n;
for(int i = 0;i < n; i++)
{
ll x;
cin >> x;
ll k = 0;
while(x) // 二进制下的每一位的1储存下来,N-t[i]就是0的个数
{
if(x&1)
XOR[k]++;
k++;
x >>= 1;
}
}
ll sum = 0;
ll k = 1;
for(ll i = 0;i < 64; i++)
{
sum = (sum % mod + k * (C(XOR[i], 3) % mod + XOR[i] % mod * C(n - XOR[i], 2) % mod) % mod) % mod;
sum %= mod;
k = (k << 1) % mod;
}
cout << sum % mod;
}
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
#ifdef FZT_ACM_LOCAL
freopen("in.txt", "r", stdin);
freopen("out.txt", "w", stdout);
#else
ios::sync_with_stdio(false);
int T = 1;
//cin >> T;
while(T--)
solve();
#endif
return 0;
}