UVALive
来源:互联网 发布:QQ浏览器MINI JAVA 编辑:程序博客网 时间:2024/06/05 09:14
题目:给你N个数,再给你M个数,问这M个数中有多少数在这N个数中出现过或可以由这N个数中的某2个数相加(可重复使用)
思路:fft
代码:
//#pragma comment(linker, "/STACK:1024000000,1024000000")#include<iostream>#include<algorithm>#include<cstdio>#include<cmath>#include<cstring>#include<string>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<list>#include<numeric>using namespace std;#define LL long long#define ULL unsigned long long#define INF 0x3f3f3f3f#define mm(a,b) memset(a,b,sizeof(a))#define PP puts("*********************");template<class T> T f_abs(T a){ return a > 0 ? a : -a; }template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}// 0x3f3f3f3f3f3f3f3f// 0x3f3f3f3fconst int maxn=1e6+50;const double PI=acos(-1.0);struct Complex{//复数结构体 double x,y; Complex(double _x=0.0,double _y=0.0){ x=_x; y=_y; } Complex operator-(const Complex &b)const{ return Complex(x-b.x,y-b.y); } Complex operator+(const Complex &b)const{ return Complex(x+b.x,y+b.y); } Complex operator*(const Complex &b)const{ return Complex(x*b.x-y*b.y,x*b.y+y*b.x); }};/**进行FFT和IFFT前的反转变换.*位置i和 (i二进制反转后位置)互换*len必须是2的幂*/void change(Complex y[],int len){ int i,j,k; for(i=1,j=len/2;i<len-1;i++){ if(i<j) swap(y[i],y[j]); k=len/2; while(j>=k){ j-=k; k/=2; } if(j<k) j+=k; }}/**做FFT*len必须为2^k形式,*on==1时是DFT,on==-1时是IDFT*/void fft(Complex y[],int len,int on){ change(y,len); for(int h=2;h<=len;h<<=1){ Complex wn(cos(-on*2*PI/h),sin(-on*2*PI/h)); for(int j=0;j<len;j+=h){ Complex w(1,0); for(int k=j;k<j+h/2;k++){ Complex u=y[k]; Complex t=w*y[k+h/2]; y[k]=u+t; y[k+h/2]=u-t; w=w*wn;//旋转因子 } } } if(on==-1) for(int i=0;i<len;i++) y[i].x/=len;}Complex x[maxn];int vis[maxn];int main(){ int N,M,dis; while(~scanf("%d",&N)){ mm(vis,0); int mx=1; for(int i=1;i<=N;i++){ scanf("%d",&dis); mx=max(mx,dis); vis[dis]=1; } int len=1; while(len<2*mx+1) len<<=1; x[0]=Complex(1,0); for(int i=1;i<len;i++) if(vis[i]) x[i]=Complex(1,0); else x[i]=Complex(0,0); fft(x,len,1); for(int i=0;i<len;i++) x[i]=x[i]*x[i]; fft(x,len,-1); for(int i=0;i<len;i++){ if(i%2==0&&vis[i/2]) x[i].x++; x[i].x/=2; } scanf("%d",&M); int ans=0; for(int i=1;i<=M;i++){ scanf("%d",&dis); if(dis>2*mx) continue; if((int)(x[dis].x+0.5)>=1) ans++; } printf("%d\n",ans); } return 0;}
阅读全文
0 0
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- UVALive
- button高亮
- php断点上传
- Jquery插件supersized.js
- JS中创建函数的几种方法
- 习题 5.16 输出星号菱形图案
- UVALive
- hbase 集群迁移:Region still in transition, waiting for it to become assigned:
- 注册头条号时怎么选择领域
- markdown编译器(自己对照学习用)
- 【OpenGL】Shader技巧集合
- 单选框 单击选中,再击取消,和 提示 选择单选框
- 公有密钥和私有密钥加密
- Android Service完全解析,关于服务你所需知道的一切(下)
- Markdown编辑器使用说明