結果

問題 No.1493 隣接xor
コンテスト
ユーザー startcpp
提出日時 2021-05-01 02:02:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 141 ms / 2,000 ms
コード長 1,829 bytes
コンパイル時間 698 ms
コンパイル使用メモリ 77,464 KB
実行使用メモリ 11,364 KB
最終ジャッジ日時 2024-07-19 05:38:59
合計ジャッジ時間 4,379 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

//累積xorを取ると、部分列は何種類？に帰着される。普通にDP考えるとO(N^2 logN)なのだが、どう高速化しようか。
#include <iostream>
#include <vector>
#include <algorithm>
#define rep(i, n) for(i = 0; i < n; i++)
using namespace std;

void radix_sort(vector<int> &a, int bit, int loopCnt) {
	for (int i = 0; i < loopCnt; i++) {
		vector<vector<int>> backet(1 << bit);
		int mask = (1 << bit) - 1;
		int shift = bit * i;
		for (int j = 0; j < a.size(); j++) {
			backet[(a[j] >> shift) & mask].push_back(a[j]);
		}
		
		int iter = 0;
		for (int j = 0; j < (1 << bit); j++) {
			for (int k = 0; k < backet[j].size(); k++) {
				a[iter] = backet[j][k];
				iter++;
			}
		}
	}
}

void press(vector<int> &a) {
	int i;
	vector<int> sa(a.size());
	rep(i, a.size()) sa[i] = a[i];
	radix_sort(sa, 15, 2);
	rep(i, a.size()) a[i] = lower_bound(sa.begin(), sa.end(), a[i]) - sa.begin();
}

int count_subarray(vector<int> a, int mod) {
	int i, j;
	
	int n = a.size();
	vector<int> dp(n + 1);		//dp[i] = (a[0],…,a[i-1])の部分列は何種類あるか
	vector<int> last(n + 1);	//last[num]：値numに関する最新のdp値.
								//より正確には、dp[i]を計算する時点で、dp[j] = num (j < i)なる最大のjにおけるdp[j]を持つ．
	int lastSum = 0;
	
	press(a);
	
	dp[0] = 1;
	for (i = 1; i <= n; i++) {
		lastSum += (dp[i - 1] - last[a[i - 1]] + mod) % mod;
		lastSum %= mod;
		last[a[i - 1]] = dp[i - 1];
		dp[i] = lastSum + 1;
		dp[i] %= mod;
	}
	
	return dp[n];
}

int main() {
	int n, i;
	
	cin >> n;
	vector<int> a(n);
	rep(i, n) cin >> a[i];
	
	vector<int> ra(n + 1);
	ra[0] = 0; rep(i, n) ra[i + 1] = ra[i] ^ a[i];
	
	vector<int> rb(n - 1);
	rep(i, n - 1) rb[i] = ra[i + 1];
	
	int mod = 1000000007;
	int ans = count_subarray(rb, mod);
	cout << ans << endl;
	return 0;
}
0