#548310 (C++14) No.1189 Sum is XOR

• 問題: No.1189 Sum is XOR
• ユーザー: toku
• 提出日時: 2020-09-05 02:42:02
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 488 ms / 2,000 ms
• コード長: 1,790 bytes
• コンパイル時間: 1,395 ms
• コンパイル使用メモリ: 168,544 KB
• 実行使用メモリ: 364,416 KB
• 最終ジャッジ日時: 2024-05-05 08:05:20
• 合計ジャッジ時間: 9,107 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	479 ms; 364,416 KB
testcase_01	AC	468 ms; 364,416 KB
testcase_02	AC	488 ms; 364,416 KB
testcase_03	AC	19 ms; 5,376 KB
testcase_04	AC	18 ms; 5,376 KB
testcase_05	AC	27 ms; 5,376 KB
testcase_06	AC	39 ms; 5,376 KB
testcase_07	AC	23 ms; 5,376 KB
testcase_08	AC	10 ms; 5,376 KB
testcase_09	AC	10 ms; 5,376 KB
testcase_10	AC	8 ms; 5,376 KB
testcase_11	AC	463 ms; 364,416 KB
testcase_12	AC	456 ms; 364,416 KB
testcase_13	AC	470 ms; 364,416 KB
testcase_14	AC	454 ms; 364,416 KB
testcase_15	AC	471 ms; 364,416 KB
testcase_16	AC	480 ms; 364,288 KB
testcase_17	AC	452 ms; 364,416 KB
testcase_18	AC	440 ms; 364,416 KB
testcase_19	AC	461 ms; 364,288 KB
testcase_20	AC	465 ms; 364,416 KB
testcase_21	AC	4 ms; 5,376 KB
testcase_22	AC	3 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>


using namespace std;
typedef long long ll;

#define REP(i,n) for(int i=0; i<int(n); i++)
#define FOR(i,m,n) for(int i=int(m); i<int(n); i++)
#define ALL(obj) (obj).begin(),(obj).end()
#define VI vector<int>
#define VP vector<pair<int,int>>
#define VPP vector<pair<int,pair<int,int>>>
#define VLL vector<long long>
#define VVI vector<vector<int>>
#define VVLL vector<vector<long long>>
#define VC vector<char>
#define VS vector<string>
#define VVC vector<vector<char>>
#define VB vector<bool>
#define VVB vector<vector<bool>>
#define fore(i,a) for(auto &i:a)
typedef pair <int, int> P;
template<typename T> using min_priority_queue = priority_queue<T, vector<T>, greater<T>>;
template<class T> bool chmax(T &a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> bool chmin(T &a, T b) { if (a > b) { a = b; return true; } return false; }

const int  INF = (1 << 30) - 1;
const ll INFL = 1LL << 60;
const ll mod = 998244353;


ll dp[2050][2050][11];


int main(){


	int n, kk;
	cin >> n >> kk;

	VI a(2050, 0);
	REP(i, n) {
		int b;
		cin >> b;
		a[b]++;
	}
	if (kk > 10) {
		cout << 0 << endl;
		return 0;
	}
	if (kk == 2) {
		ll ans = 0;
		REP(i, 2050) {
			FOR(j, i + 1, 2050) {
				if ((i + j) == (i^j)) {
					ans += a[i] * a[j];
					ans %= mod;
				}
			}
		}
		cout << ans << endl;
		return 0;
	}

	dp[0][0][0] = 1;

	FOR(i, 1, 2050) {
		REP(j, 2050)REP(k, kk) {
			dp[i][j][k] += dp[i - 1][j][k];
			dp[i][j][k] %= mod;
		}
		if (a[i] == 0)continue;
		REP(j, 2050) {
			if ((i + j) != (i^j))continue;
			REP(k, kk) {
				dp[i][i + j][k + 1] += dp[i - 1][j][k] * a[i];
				dp[i][i + j][k + 1] %= mod;
			}
		}
	}
	ll ans = 0;
	REP(i, 2050) {
		REP(j, 2050) {
			ans += dp[i][j][kk];
			ans %= mod;
		}
	}
	cout << ans << endl;



}