#558344 (C++17) No.1240 Or Sum of Xor Pair

提出ソース

結果

問題	No.1240 Or Sum of Xor Pair
ユーザー	n_vip
提出日時	2020-09-25 22:35:34
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	208 ms / 2,000 ms
コード長	4,020 bytes
コンパイル時間	1,330 ms
コンパイル使用メモリ	136,136 KB
実行使用メモリ	10,876 KB
最終ジャッジ日時	2023-09-10 15:53:36
合計ジャッジ時間	8,445 ms
ジャッジサーバーID （参考情報）	judge11 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	138 ms 9,196 KB
testcase_01	AC	139 ms 9,136 KB
testcase_02	AC	137 ms 9,408 KB
testcase_03	AC	138 ms 9,248 KB
testcase_04	AC	137 ms 9,100 KB
testcase_05	AC	138 ms 9,240 KB
testcase_06	AC	136 ms 9,224 KB
testcase_07	AC	138 ms 9,216 KB
testcase_08	AC	138 ms 9,128 KB
testcase_09	AC	139 ms 9,364 KB
testcase_10	AC	144 ms 9,444 KB
testcase_11	AC	145 ms 9,312 KB
testcase_12	AC	149 ms 9,692 KB
testcase_13	AC	148 ms 9,376 KB
testcase_14	AC	149 ms 9,372 KB
testcase_15	AC	203 ms 10,744 KB
testcase_16	AC	205 ms 10,780 KB
testcase_17	AC	204 ms 10,864 KB
testcase_18	AC	204 ms 10,780 KB
testcase_19	AC	206 ms 10,656 KB
testcase_20	AC	204 ms 10,876 KB
testcase_21	AC	205 ms 10,636 KB
testcase_22	AC	205 ms 10,808 KB
testcase_23	AC	206 ms 10,800 KB
testcase_24	AC	204 ms 10,660 KB
testcase_25	AC	208 ms 10,640 KB
testcase_26	AC	204 ms 10,764 KB
testcase_27	AC	138 ms 9,060 KB
testcase_28	AC	137 ms 9,188 KB
testcase_29	AC	166 ms 10,768 KB
testcase_30	AC	153 ms 9,844 KB
testcase_31	AC	159 ms 9,880 KB
testcase_32	AC	166 ms 10,800 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <string>
#include <vector>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<stack>
#include<queue>
#include<cmath>
#include<algorithm>
#include<functional>
#include<list>
#include<deque>
#include<bitset>
#include<set>
#include<map>
#include<unordered_map>
#include<unordered_set>
#include<cstring>
#include<sstream>
#include<complex>
#include<iomanip>
#include<numeric>
#include<cassert>
#include<random>
#define X first
#define Y second
#define pb push_back
#define rep(X,Y) for (int (X) = 0;(X) < (int)(Y);++(X))
#define reps(X,S,Y) for (int (X) = (int)(S);(X) < (int)(Y);++(X))
#define rrep(X,Y) for (int (X) = (int)(Y)-1;(X) >=0;--(X))
#define rreps(X,S,Y) for (int (X) = (int)(Y)-1;(X) >= (int)(S);--(X))
#define repe(X,Y) for ((X) = 0;(X) < (Y);++(X))
#define peat(X,Y) for (;(X) < (Y);++(X))
#define all(X) (X).begin(),(X).end()
#define rall(X) (X).rbegin(),(X).rend()
#define eb emplace_back
#define UNIQUE(X) (X).erase(unique(all(X)),(X).end())
#define Endl endl
#define NL <<"\n"
#define cauto const auto

using namespace std;
using ll=long long;
using pii=pair<int,int>;
using pll=pair<ll,ll>;
template<class T> using vv=vector<vector<T>>;
template<class T> inline bool MX(T &l,const T &r){return l<r?l=r,1:0;}
template<class T> inline bool MN(T &l,const T &r){return l>r?l=r,1:0;}
//#undef NUIP
#ifdef NUIP
#include "benri.h"
#else
#define out(args...)
#endif
#ifdef __cpp_init_captures
template<typename T>vector<T> table(int n, T v){ return vector<T>(n, v);}
template <class... Args> auto table(int n, Args... args){auto val = table(args...); return vector<decltype(val)>(n, move(val));}
#endif
template<class A,class B> pair<A,B> operator+(const pair<A,B> &p,const pair<A,B> &q){ return {p.X+q.X,p.Y+q.Y};}
template<class A,class B,class C,class D> pair<A,B>& operator+=(pair<A,B> &p,const pair<C,D> &q){ p.X+=q.X; p.Y+=q.Y; return p;}
template<class A,class B> pair<A,B> operator-(const pair<A,B> &p,const pair<A,B> &q){ return {p.X-q.X,p.Y-q.Y};}
template<class A,class B,class C,class D> pair<A,B>& operator-=(pair<A,B> &p,const pair<C,D> &q){ p.X-=q.X; p.Y-=q.Y; return p;}
template<class A,class B> istream& operator>>(istream &is, pair<A,B> &p){ is>>p.X>>p.Y; return is;}
template<class T=ll> T read(){ T re; cin>>re; return move(re);}
template<class T=ll> T read(const T &dec){ T re; cin>>re; return re-dec;}
template<class T=ll> vector<T> readV(const int sz){ vector<T> re(sz); for(auto &x:re) x=read<T>(); return move(re);}
template<class T=ll> vector<T> readV(const int sz, const T &dec){ vector<T> re(sz); for(auto &x:re) x=read<T>(dec); return move(re);}
vv<int> readG(const int &n,const int &m){ vv<int> g(n); rep(_,m){ cauto a=read<int>(1),b=read<int>(1); g[a].pb(b); g[b].pb(a);} return move(g);}
vv<int> readG(const int &n){ return readG(n,n-1);}
const ll MOD=1e9+7; //998244353

void FWT(ll *a,int n) {
  for (int d=1; d<n; d*=2) {
    int m = d*2;
    for (int i=0; i<n; i+=m) {
      rep(j, d) {
        ll x=a[i+j],y=a[i+j+d];
        a[i+j]=x+y, a[i+j+d]=x-y;
      }
    }
  }
}

void UFWT(ll *a,int n){
  for(int d=1; d<n; d*=2) {
    int m = d*2;
    for(int i=0; i<n; i+=m) {
      rep(j, d) {
        ll x=a[i+j],y=a[i+j+d];
        a[i+j]=(x+y)/2, a[i+j+d]=(x-y)/2;
      }
    }
  }
}

int main(){
  ios_base::sync_with_stdio(false); cin.tie(0);
  cout<<fixed<<setprecision(0);
	cauto n=read();
	cauto x=read();
	cauto vs=readV(n);

	vector<ll> a(1<<18);
	rep(i,x) a[i]=1;
	FWT(&a[0],a.size());

	vector<ll> cnt(1<<18);
	for(cauto v:vs) cnt[v]+=1;
	FWT(&cnt[0],cnt.size());
	for(auto &x:cnt) x*=x;
	UFWT(&cnt[0],cnt.size());
	cauto prs=(accumulate(cnt.begin(),cnt.begin()+x,0ll)-0);
	out(prs,1);

	ll re=0;
	rep(d,18){
		vector<ll> c(1<<18);
		for(cauto v:vs)if(~v>>d&1) c[v]+=1;
		FWT(&c[0],c.size());
		rep(i,c.size()) c[i]*=a[i];
		UFWT(&c[0],c.size());
		// rep(i,10) cout<<c[i]<<","; cout NL;
		auto rem=prs;
		for(cauto v:vs)if(~v>>d&1) rem-=c[v];
		out(d,rem,1);
		re+=rem<<d;
	}

	out(re,1);
	for(cauto v:vs) re-=v;
	cout<<re/2 NL;
  return 0;
}

yukicoder

結果

テストケース

ソースコード