結果

問題 No.1240 Or Sum of Xor Pair
ユーザー 沙耶花沙耶花
提出日時 2021-10-14 20:08:35
言語 C++17
(gcc 11.1.0 + boost 1.77.0)
結果
AC  
実行時間 546 ms / 2,000 ms
コード長 1,475 Byte
コンパイル時間 4,079 ms
使用メモリ 9,640 KB
最終ジャッジ日時 2021-10-14 20:09:00
合計ジャッジ時間 22,607 ms
ジャッジサーバーID
(参考情報)
judge12 / judge15
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 430 ms
5,576 KB
testcase_01 AC 437 ms
5,652 KB
testcase_02 AC 461 ms
5,584 KB
testcase_03 AC 444 ms
5,608 KB
testcase_04 AC 441 ms
5,736 KB
testcase_05 AC 438 ms
7,020 KB
testcase_06 AC 462 ms
7,020 KB
testcase_07 AC 436 ms
7,020 KB
testcase_08 AC 455 ms
5,744 KB
testcase_09 AC 463 ms
7,020 KB
testcase_10 AC 444 ms
7,020 KB
testcase_11 AC 438 ms
6,208 KB
testcase_12 AC 458 ms
6,380 KB
testcase_13 AC 451 ms
6,416 KB
testcase_14 AC 454 ms
6,308 KB
testcase_15 AC 541 ms
9,484 KB
testcase_16 AC 528 ms
9,524 KB
testcase_17 AC 515 ms
9,496 KB
testcase_18 AC 527 ms
9,584 KB
testcase_19 AC 529 ms
9,640 KB
testcase_20 AC 520 ms
9,536 KB
testcase_21 AC 539 ms
9,568 KB
testcase_22 AC 515 ms
9,492 KB
testcase_23 AC 514 ms
9,504 KB
testcase_24 AC 512 ms
9,508 KB
testcase_25 AC 535 ms
9,476 KB
testcase_26 AC 501 ms
9,500 KB
testcase_27 AC 429 ms
5,660 KB
testcase_28 AC 422 ms
7,020 KB
testcase_29 AC 546 ms
9,040 KB
testcase_30 AC 482 ms
6,808 KB
testcase_31 AC 441 ms
7,284 KB
testcase_32 AC 501 ms
8,656 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 100000000000000000

//Xor

	template <typename T>
	void fwt(vector<T>& f) {
		int n = f.size();
		for (int i = 1; i < n; i <<= 1) {
			for (int j = 0; j < n; j++) {
				if ((j & i) == 0) {
					T x = f[j], y = f[j | i];
					f[j] = x + y, f[j | i] = x - y;
				}
			}
		}
	}
	template <typename T>
	void ifwt(vector<T>& f) {
		int n = f.size();
		for (int i = 1; i < n; i <<= 1) {
			for (int j = 0; j < n; j++) {
				if ((j & i) == 0) {
					T x = f[j], y = f[j | i];
					f[j] = (x + y) / 2, f[j | i] = (x - y) / 2;
				}
			}
		}
	}


long long get(vector<int> a,long long X){
	vector<long long> dp(1<<18,0);
	rep(i,a.size())dp[a[i]]++;
	fwt(dp);
	rep(i,dp.size())dp[i] *= dp[i];
	ifwt(dp);
	long long ret = 0LL;
	rep(i,X){
		ret += dp[i];
	}
	ret -= a.size();
	ret /= 2;
	return ret;
	
}

int main(){
	
	int N,X;
	cin>>N>>X;
	
	vector<int> a(N);
	rep(i,N){
		scanf("%d",&a[i]);
	}
	long long ans = 0LL;
	rep(i,18){
		int t = 1<<18;
		t--;
		if((X>>i)&1){
			t ^= 1<<i;
		}
		else{
			for(int j=0;j<=i;j++)t ^= 1<<j;
		}
		vector<vector<int>> dp(2);
		rep(j,N){
			dp[(a[j]>>i)&1].push_back(a[j]);
		}
		long long sum = 0;
		sum -= get(dp[0],X);
		rep(j,dp[1].size())dp[0].push_back(dp[1][j]);
		sum += get(dp[0],X);
		ans += sum<<i;
		
	}
	cout<<ans<<endl;
	return 0;
}
0