結果

問題 No.206 数の積集合を求めるクエリ
ユーザー smiken_61smiken_61
提出日時 2018-10-31 19:26:57
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 234 ms / 7,000 ms
コード長 13,893 bytes
コンパイル時間 1,910 ms
コンパイル使用メモリ 169,536 KB
実行使用メモリ 38,596 KB
最終ジャッジ日時 2024-11-19 12:43:57
合計ジャッジ時間 7,041 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 82 ms
34,816 KB
testcase_01 AC 82 ms
36,868 KB
testcase_02 AC 79 ms
35,576 KB
testcase_03 AC 83 ms
38,596 KB
testcase_04 AC 81 ms
37,064 KB
testcase_05 AC 78 ms
35,780 KB
testcase_06 AC 84 ms
37,404 KB
testcase_07 AC 84 ms
36,824 KB
testcase_08 AC 83 ms
37,640 KB
testcase_09 AC 83 ms
37,468 KB
testcase_10 AC 82 ms
35,660 KB
testcase_11 AC 81 ms
37,320 KB
testcase_12 AC 87 ms
37,788 KB
testcase_13 AC 86 ms
36,244 KB
testcase_14 AC 90 ms
37,400 KB
testcase_15 AC 85 ms
35,208 KB
testcase_16 AC 84 ms
36,748 KB
testcase_17 AC 110 ms
36,232 KB
testcase_18 AC 97 ms
35,992 KB
testcase_19 AC 101 ms
36,916 KB
testcase_20 AC 90 ms
34,552 KB
testcase_21 AC 98 ms
37,308 KB
testcase_22 AC 96 ms
35,852 KB
testcase_23 AC 106 ms
36,444 KB
testcase_24 AC 234 ms
36,512 KB
testcase_25 AC 230 ms
36,852 KB
testcase_26 AC 220 ms
35,284 KB
testcase_27 AC 179 ms
36,448 KB
testcase_28 AC 219 ms
35,344 KB
testcase_29 AC 224 ms
36,392 KB
testcase_30 AC 219 ms
36,440 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

     #include <bits/stdc++.h>

//    #include <boost/multiprecision/cpp_int.hpp>
 #define int long long
 #define inf  1000000007
 #define pa pair<int,int>
 #define ll long long
 #define pal pair<double,double>
 #define ppap pair<pa,int>
  #define PI 3.14159265358979323846
  #define paa pair<int,char>
  #define  mp make_pair
  #define  pb push_back
  #define EPS (1e-10)
                                          
    int dx[8]={0,1,0,-1,1,1,-1,-1};
    int dy[8]={1,0,-1,0,-1,1,1,-1};
                                            using namespace std;
                                   			class pa3{
                                            	public:
                                            	int x,y,z;
                                            	pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
                                            	bool operator < (const pa3 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		 return z<p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                   				bool operator > (const pa3 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		 return z>p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa3 &p) const{
                                            		return x==p.x && y==p.y && z==p.z;
                                            	}
                                            		bool operator != (const pa3 &p) const{
                                            			return !( x==p.x && y==p.y && z==p.z);
                                            	}
                                            
                                            };
                                            
                                            class pa4{
                                            	public:
                                            	int x;
                                            	int y,z,w;
                                            	pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
                                            	bool operator < (const pa4 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		if(z!=p.z)return z<p.z;
                                            		return w<p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator > (const pa4 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		if(z!=p.z)return z>p.z;
                                            		return w>p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa4 &p) const{
                                            		return x==p.x && y==p.y && z==p.z &&w==p.w;
                                            	}
                                            		
                                            
                                            };
                                            class pa2{
                                            	public:
                                            	int x,y;
                                            	pa2(int x=0,int y=0):x(x),y(y) {}
                                            	pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
                                            	pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
                                            	bool operator < (const pa2 &p) const{
                                            		return y != p.y ? y<p.y: x<p.x;
                                            	}
                                            	bool operator > (const pa2 &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa2 &p) const{
                                            		return abs(x-p.x)==0 && abs(y-p.y)==0;
                                            	}
                                            	bool operator != (const pa2 &p) const{
                                            		return !(abs(x-p.x)==0 && abs(y-p.y)==0);
                                            	}
                                            		
                                            
                                            };
                                            
/*
                                            class Point{
                                            	public:
                                            	double x,y;
                                            	Point(double x=0,double y=0):x(x),y(y) {}
                                            	Point operator + (Point p) {return Point(x+p.x,y+p.y);}
                                            	Point operator - (Point p) {return Point(x-p.x,y-p.y);}
                                            	Point operator * (double a) {return Point(x*a,y*a);}
                                            	Point operator / (double a) {return Point(x/a,y/a);}
                                            	double absv() {return sqrt(norm());}
                                            	double norm() {return x*x+y*y;}
                                            	bool operator < (const Point &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const Point &p) const{
                                            		return fabs(x-p.x)<EPS && fabs(y-p.y)<EPS;
                                            	}
                                            };
                                            typedef Point Vector;
                                     #define pl pair<int,pas>
                                            struct Segment{
                                            Point p1,p2;
                                            };
                                             double dot(Vector a,Vector b){
                                            	return a.x*b.x+a.y*b.y;
                                            }
                                            double cross(Vector a,Vector b){
                                            	return a.x*b.y-a.y*b.x;
                                            }
                                        
                bool parareru(Point a,Point b,Point c,Point d){
                //	if(abs(cross(a-b,d-c))<EPS)cout<<"dd "<<cross(a-b,d-c)<<endl;
                	return abs(cross(a-b,d-c))<EPS;
                }
                double distance_ls_p(Point a, Point b, Point c) {
                  if ( dot(b-a, c-a) < EPS ) return (c-a).absv();
                  if ( dot(a-b, c-b) < EPS ) return (c-b).absv();
                  return abs(cross(b-a, c-a)) / (b-a).absv();
                }
                bool is_intersected_ls(Segment a,Segment b) {
                	if(a.p1==b.p1||a.p2==b.p1||a.p1==b.p2||a.p2==b.p2) return false;
                	if(parareru((a.p2),(a.p1),(a.p1),(b.p2))&&parareru((a.p2),(a.p1),(a.p1),(b.p1))){
                //		cout<<"sss"<<endl;
                		if(dot(a.p1-b.p1,a.p1-b.p2)<EPS) return true;
                		if(dot(a.p2-b.p1,a.p2-b.p2)<EPS) return true;
                		if(dot(a.p1-b.p1,a.p2-b.p1)<EPS) return true;
                		if(dot(a.p1-b.p2,a.p2-b.p2)<EPS) return true;
                		return false;
                	}
                  else return ( cross(a.p2-a.p1, b.p1-a.p1) * cross(a.p2-a.p1, b.p2-a.p1) < EPS ) && ( cross(b.p2-b.p1, a.p1-b.p1) * cross(b.p2-b.p1, a.p2-b.p1) < EPS );
                }
                 
                double segment_dis(Segment a,Segment b){
                	if(is_intersected_ls(a,b))return 0;
                	double r=distance_ls_p(a.p1, a.p2, b.p1);
                	r=min(r,distance_ls_p(a.p1, a.p2, b.p2));
                	r=min(r,distance_ls_p(b.p1, b.p2, a.p2));
                	r=min(r,distance_ls_p(b.p1, b.p2, a.p1));
                	return r;
                }
                Point intersection_ls(Segment a, Segment b) {
                  Point ba = b.p2-b.p1;
                  double d1 = abs(cross(ba, a.p1-b.p1));
                  double d2 = abs(cross(ba, a.p2-b.p1));
                  double t = d1 / (d1 + d2);
                 
                  return a.p1 + (a.p2-a.p1) * t;
                }
 */                
                                string itos( int i ) {
                                ostringstream s ;
                                s << i ;
                                return s.str() ;
                                }
                                 
                                int gcd(int v,int b){
                                	if(v>b) return gcd(b,v);
                                	if(v==b) return b;
                                	if(b%v==0) return v;
                                	return gcd(v,b%v);
                                }
                 
                                double distans(double x1,double y1,double x2,double y2){
                                	double rr=(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
                                	return sqrt(rr);
                                	
                                }
                                int mod;
                                
                int pr[500010];
                int inv[500010];
                
                int beki(int wa,int rr,int warukazu){
                	if(rr==0) return 1%warukazu;
                	if(rr==1) return wa%warukazu;
                	wa%=warukazu;
                	if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
                	ll zx=beki(wa,rr/2,warukazu);
                	return (zx*zx)%warukazu;
                }
    double bekid(double w,int r){
    	if(r==0) return 1.0;
    	if(r==1) return w;
    	if(r%2) return bekid(w,r-1)*w;
    	double f=bekid(w,r/2);
    	return f*f;
    }
                
    			int comb(int nn,int rr){
    				int r=pr[nn]*inv[rr];
    				r%=mod;
    				r*=inv[nn-rr];
    				r%=mod;
    				return r;
    			}
                /*
                void gya(int ert){
                	pr[0]=1;
                	for(int i=1;i<ert;i++){
                		pr[i]=(pr[i-1]*i)%mod;
                	}
                	for(int i=0;i<ert;i++) inv[i]=beki(pr[i],mod-2,mod);
                	
                }
                */
              //   cin.tie(0);
    		//	ios::sync_with_stdio(false);
    			//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;            
                 //sort(ve.begin(),ve.end(),greater<int>());
                                         //----------------kokomade tenpure------------
                //vector<double> ans(100000000),ans2(100000000)

class comp{
public:
double re,im;
comp(double re=0,double im=0):re(re),im(im) {};
comp operator + (comp p) {return comp(re+p.re,im+p.im);}	
comp operator - (comp p) {return comp(re-p.re,im-p.im);}
comp operator * (comp p) {return comp(re*p.re-im*p.im,re*p.im+im*p.re);}
comp operator / (double p){return comp(re/p,im/p);}
 
};
comp D[1000000];
void fourie(int N,int d,vector<comp> &vec ){// 1<<d == N
	// Uは順方向ではexp(2pi*I/N)逆だとexp(-2pi*I/N)
	if(N==1){
		return;
	}
	int gyaku=0;
	for(int i=1;i<N;i++){
		int e=1<<(d-1);
		while(gyaku&e){
			gyaku^=e;
			e>>=1;
		}
		gyaku^=e;
		if(i<gyaku)swap(vec[i],vec[gyaku]);	
	}
	for(int c=0;c<d;c++){
		int f=1<<c;
		for(int i=0;i<N;i+=(f<<1)){
			for(int j=i;j<i+f;j++){
				comp A1=vec[j]+vec[j+f]*D[(j-i)<<(d-c-1)];
				comp A2=vec[j]-vec[j+f]*D[(j-i)<<(d-c-1)];
				vec[j]=A1;
				vec[j+f]=A2;
			}
		}
	}
 
	return;
}
 
vector<int> FFT(vector<comp> input1,vector<comp> input2){
int N=1;
	int d=0;
	int size1=input1.size(),size2=input2.size();
	while(N<size1+size2){
		N*=2;
		d++;
	}
	assert(N==(1<<d));
	while(input1.size()<N)input1.pb(0);
	while(input2.size()<N)input2.pb(0);
//	complex<double> un=exp(complex<double>(0,2*PI/N));
	for(int i=0;i<N;i++){
		
		complex<double> un=exp(complex<double>(0.0,(double)2*PI*i/N));
		D[i]=(comp){un.real(),un.imag()};
	}
	fourie(N,d,input1);
	fourie(N,d,input2);
	vector<int> ANS;
	
	for(int i=0;i<N;i++)input1[i]=input1[i]*input2[i],(D[i].im)*=-1;
	
	fourie(N,d,input1);
	for(int i=0;i<N;i++)input1[i]=input1[i]/(N*1.0);
	
	for(int i=0;i<N;i++){
		ANS.pb((int) (0.5+input1[i].re));
	//	cout<<ANS.back()<<endl;
	//	printf("%.10lf\n", INV[i].real());
	}
	return ANS;
}
 

  signed main(){
    	       cin.tie(0);
    			ios::sync_with_stdio(false);
int l,m,n;
cin>>l>>m>>n;
  		vector<comp> ve1(100001,{0,0});
	vector<comp> ve2(100001,{0,0});
  
  	for(int i=0;i<l;i++){
  		int y;
  		cin>>y;
  		ve1[y].re=1;
  	}
  	
  	
  	for(int i=0;i<m;i++){
  		int y;
  		cin>>y;
  		ve2[y].re=1;
  	}
  	reverse(ve2.begin(),ve2.end());
  	vector<int> ve=FFT(ve1,ve2);
  	int q;
  	cin>>q;
  	for(int i=0;i<q;i++)cout<<ve[100000+i]<<endl;
  	return 0;
    }
    

0