結果

問題 No.206 数の積集合を求めるクエリ
ユーザー startcppstartcpp
提出日時 2015-03-30 17:25:42
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 980 ms / 7,000 ms
コード長 2,688 bytes
コンパイル時間 675 ms
コンパイル使用メモリ 65,948 KB
実行使用メモリ 5,380 KB
最終ジャッジ日時 2023-09-17 00:23:26
合計ジャッジ時間 9,093 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 3 ms
4,380 KB
testcase_02 AC 3 ms
4,376 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 3 ms
4,380 KB
testcase_05 AC 3 ms
4,380 KB
testcase_06 AC 3 ms
4,376 KB
testcase_07 AC 3 ms
4,380 KB
testcase_08 AC 3 ms
4,376 KB
testcase_09 AC 2 ms
4,376 KB
testcase_10 AC 2 ms
4,376 KB
testcase_11 AC 2 ms
4,380 KB
testcase_12 AC 4 ms
4,376 KB
testcase_13 AC 4 ms
4,380 KB
testcase_14 AC 4 ms
4,376 KB
testcase_15 AC 3 ms
4,380 KB
testcase_16 AC 4 ms
4,376 KB
testcase_17 AC 23 ms
5,208 KB
testcase_18 AC 13 ms
4,536 KB
testcase_19 AC 22 ms
5,028 KB
testcase_20 AC 14 ms
4,376 KB
testcase_21 AC 16 ms
4,596 KB
testcase_22 AC 16 ms
4,604 KB
testcase_23 AC 22 ms
5,276 KB
testcase_24 AC 979 ms
5,380 KB
testcase_25 AC 957 ms
5,012 KB
testcase_26 AC 980 ms
4,616 KB
testcase_27 AC 700 ms
4,380 KB
testcase_28 AC 966 ms
4,664 KB
testcase_29 AC 968 ms
4,668 KB
testcase_30 AC 938 ms
4,408 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<vector>
#define DB(x) 
using namespace std;

class AandBProblem {
private:
	int L, M, N;
	int A[100000], B[100000];
	int Q;

public:
	void input(int l, int m, int n, int *a, int *b, int q) {
		L = l;
		M = m;
		N = n;
		for(int i = 0; i < l; i++) A[i] = a[i];
		for(int i = 0; i < m; i++) B[i] = b[i];
		Q = q;
	}

	//O(N(L+M)Q)
	vector<int> solve1() {
		int i, j, k;
		vector<int> ret;
		ret.resize(Q);
		for( i = 0; i < Q; i++ ) {
			ret[i] = 0;
			for( j = 1; j <= N; j++ ) {
				for( k = 0; k < L; k++ )
					if ( A[k] == j ) break;
				if ( k == L ) continue;
				for( k = 0; k < M; k++ )
					if ( B[k] + i == j ) break;
				if ( k == M ) continue;
				ret[i]++;
			}
		}
		return ret;
	}
	//O(NQ)の定数1なやつ。
	vector<int> solve2() {
		int i, j;
		vector<int> ret;
		ret.resize(Q);
		for( i = 0; i < Q; i++ ) {
			ret[i] = 0;
			vector<int> paket;
			paket.resize(N+1+i);
			for( j = 0; j < N+1+i; j++ ) paket[j] = 0;
			
			for( j = 0; j < L; j++ ) {
				paket[A[j]]++;
			}
			for( j = 0; j < M; j++ ) {
				paket[B[j]+i]++;
			}

			for( j = 0; j < N+1; j++ ) {
				if ( paket[j] == 2 )
					ret[i]++;
			}
		}
		return ret;
	}

	//O(LM)の定数ちょっと速いやつ
	vector<int> solve3() {
		int i, j;
		int ans[100001] = {0};
		sort(A, A+L);
		sort(B, B+M);

		for( i = 0; i < L; i++ ) {
			for( j = 0; j < M; j++ ) {
				if ( B[j] > A[i] ) break;
				ans[ A[i] - B[j] ]++;
			}
		}

		vector<int> ret;
		for( i = 0; i < Q; i++ )
			ret.push_back(ans[i]);
		return ret;
	}

	//想定解：ビットで畳み込んで定数高速化したO(NQ)解法
	vector<int> solve4() {
		unsigned short int uA[8000] = {0}, uB[8000] = {0};
		short int bcnt[65536] = {0};

		for(int i = 0; i < L; i++ )
			uA[A[i]/16] += (1 << (A[i]%16));
		for(int i = 0; i < M; i++ )
			uB[B[i]/16] += (1 << (B[i]%16));

		for(int i = 0; i < 65536; i++ ) {
			for(int j = 0; j < 16; j++ )
				bcnt[i] += (i>>j)&1;
		}

		short int n = N/16 + 3;
		vector<int> ans;
		ans.resize(Q, 0);
		for(int i = 0; i < Q; i++ ) {
			for( short int j = 0; j < n; j++ ) {
				ans[i] += bcnt[uA[j]&uB[j]];
			}
			for( short int j = n - 1; j >= 0; j-- ) {
				uB[j+1] += (uB[j] >> 15);
				uB[j] = (uB[j] << 1);
			}
		}
		return ans;
	}

}test;

int L, M, N;
int A[100000], B[100000];
int Q;

int main() {
	scanf("%d%d%d", &L, &M, &N);
	for(int i = 0; i < L; i++ ) scanf("%d", A+i);
	for(int i = 0; i < M; i++ ) scanf("%d", B+i);
	scanf("%d", &Q);
	test.input(L, M, N, A, B, Q);

	//printf("Test\n");

	//vector<int> res2 = test.solve2();
	vector<int> res4 = test.solve4();

	for(int i = 0; i < Q; i++ )
		printf("%d\n", res4[i]);
	return 0;
}
0