#25618 (C++11) No.206 数の積集合を求めるクエリ

• 問題: No.206 数の積集合を求めるクエリ
• ユーザー: 古寺いろは
• 提出日時: 2015-05-08 23:55:08
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 664 ms / 7,000 ms
• コード長: 1,436 bytes
• コンパイル時間: 1,412 ms
• コンパイル使用メモリ: 164,824 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-05 20:50:56
• 合計ジャッジ時間: 11,147 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	4 ms; 5,376 KB
testcase_07	AC	4 ms; 5,376 KB
testcase_08	AC	4 ms; 5,376 KB
testcase_09	AC	4 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	9 ms; 5,376 KB
testcase_13	AC	7 ms; 5,376 KB
testcase_14	AC	8 ms; 5,376 KB
testcase_15	AC	7 ms; 5,376 KB
testcase_16	AC	8 ms; 5,376 KB
testcase_17	AC	536 ms; 5,376 KB
testcase_18	AC	527 ms; 5,376 KB
testcase_19	AC	537 ms; 5,376 KB
testcase_20	AC	500 ms; 5,376 KB
testcase_21	AC	505 ms; 5,376 KB
testcase_22	AC	501 ms; 5,376 KB
testcase_23	AC	531 ms; 5,376 KB
testcase_24	AC	664 ms; 5,376 KB
testcase_25	AC	660 ms; 5,376 KB
testcase_26	AC	642 ms; 5,376 KB
testcase_27	AC	585 ms; 5,376 KB
testcase_28	AC	631 ms; 5,376 KB
testcase_29	AC	652 ms; 5,376 KB
testcase_30	AC	621 ms; 5,376 KB

ソースコード

#include "bits/stdc++.h"
using namespace std;

int bitcount(long long bits){
	bits = (bits & 0x5555555555555555) + (bits >> 1 & 0x5555555555555555);
	bits = (bits & 0x3333333333333333) + (bits >> 2 & 0x3333333333333333);
	bits = (bits & 0x0f0f0f0f0f0f0f0f) + (bits >> 4 & 0x0f0f0f0f0f0f0f0f);
	bits = (bits & 0x00ff00ff00ff00ff) + (bits >> 8 & 0x00ff00ff00ff00ff);
	bits = (bits & 0x0000ffff0000ffff) + (bits >> 16 & 0x0000ffff0000ffff);
	return (bits & 0x00000000ffffffff) + (bits >> 32 & 0x00000000ffffffff);
}

int main() {
	int L, M, N;
	cin >> L >> M >> N;
	vector<int> A(L);
	vector<int> B(M);
	vector<int> Bpos(N + 10000);
	for (int i = 0; i < L; i++)
	{
		cin >> A[i];
		A[i]--;
	}
	for (int i = 0; i < M; i++)
	{
		cin >> B[i];
		B[i]--;
		Bpos[B[i]]++;
	}
	sort(A.begin(), A.end());
	sort(B.begin(), B.end());

	int Q;
	cin >> Q;

	vector<int> ans(N + Q + 100, 0);
	for (int j = 0; j < 60; j++)
	{
		vector<long long> al(N / 60 + 100);
		for (int i = 0; i < A.size(); i++)
		{
			if (A[i] < j) continue;
			al[(A[i] - j) / 60] |= 1ll << ((A[i] - j) % 60);
		}

		for (int i = 0; i < (N - j + 59) / 60; i++)
		{
			long long bit = 0;
			for (int k = 0; k < 60; k++)
			{
				if (Bpos[i * 60 + k] == 1){
					bit |= 1ll << k;
				}
			}
			for (int k = i; k < (N - j + 59) / 60; k++)
			{
				ans[j + (k - i) * 60] += bitcount(al[k] & bit);
			}
		}
	}
	for (int i = 0; i < Q; i++)
	{
		cout << ans[i] << endl;
	}
	cin >> N;

}