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

• 問題: No.1189 Sum is XOR
• ユーザー: toku
• 提出日時: 2020-09-05 02:39:50
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,769 bytes
• コンパイル時間: 1,441 ms
• コンパイル使用メモリ: 168,612 KB
• 実行使用メモリ: 364,544 KB
• 最終ジャッジ日時: 2024-05-05 08:02:07
• 合計ジャッジ時間: 9,468 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	WA	-
testcase_03	AC	18 ms; 5,376 KB
testcase_04	AC	17 ms; 5,376 KB
testcase_05	AC	27 ms; 5,376 KB
testcase_06	AC	38 ms; 5,376 KB
testcase_07	AC	22 ms; 5,376 KB
testcase_08	AC	10 ms; 5,376 KB
testcase_09	AC	10 ms; 5,376 KB
testcase_10	AC	9 ms; 5,376 KB
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	AC	5 ms; 5,376 KB
testcase_22	AC	5 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, 11) {
			dp[i][j][k] += dp[i - 1][j][k];
			dp[i][j][k] %= mod;
		}

		REP(j, 2050 - i) {
			if ((i + j) != (i^j))continue;
			REP(k, 10) {
				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;



}