小Q与进位制 - 分治FFT

题目大意:有一种进制,第i位的进位值是 basei b a s e i ,然后现在给你一个数字a,问有多少数字不超过a,保证a有n位且最高位不为0, n120000 n ≤ 120000
题解:答案是: ni=1aijibj ∑ i = 1 n a i ∏ j ≤ i b j ,考虑将a和b的每一项理解为多项式去做分治法法塔,像维护哈希一样维护b的乘积和答案即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
#include<assert.h>
#define lint long long
#define db long double
#define gc getchar()
#define N 130000
#define BAS 100000//10000000
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
const db PI=acos(-1);
inline int inn()
{
    int x,ch;while((ch=gc)<'0'||ch>'9');
    x=ch^'0';while((ch=gc)>='0'&&ch<='9')
        x=(x<<1)+(x<<3)+(ch^'0');return x;
}
struct E{
    db x,y;E(db _x=0,db _y=0) { x=_x,y=_y; }
    inline E operator+(const E &b)const { return E(x+b.x,y+b.y); }
    inline E operator+=(const E &b) { return (*this)=(*this)+b; }
    inline E operator-(const E &b)const { return E(x-b.x,y-b.y); }
    inline E operator*(const E &b)const { return E(x*b.x-y*b.y,x*b.y+y*b.x); }
    inline E operator*=(const E &b) { return (*this)=(*this)*b; }
};int r[N<<4];vector<E> a[N],b[N],t1,t2;
inline int show(vector<E> &a,int t=1)//type = 0 : show number;else show polygon
{
    int n=(int)a.size();
    if(!t)
    {
        printf("%d",(int)a[n-1].x);
        for(int i=n-2;i>=0;i--)
            printf("%05d",(int)a[i].x);//
        return printf("\n"),0;
    }
    for(int i=0;i<n;i++) cerr<<(int)a[i].x sp;cerr ln;return 0;
}
inline int FFT(vector<E> &a,int n,int sgn)
{
    for(int i=1;i<n;i++) if(i>r[i]) swap(a[i],a[r[i]]);
    for(int i=1;i<n;i<<=1)
    {
        E wn(cos(PI/i),sgn*sin(PI/i));
        for(int j=0,t=i<<1;j<n;j+=t)
        {
            E w(1,0);
            for(int k=0;k<i;k++,w*=wn)
            {
                E x=a[j+k],y=w*a[j+k+i];
                a[j+k]=x+y,a[j+k+i]=x-y;
            }
        }
    }
    if(sgn<0) for(int i=0;i<n;i++) a[i].x/=n,a[i].y/=n;return 0;
}
inline int tms(vector<E> &a,vector<E> &b,vector<E> &c)
{
    int m1=(int)a.size(),m2=(int)b.size(),n=1,L=0;for(;n<m1+m2-1;n<<=1,L++);
    for(int i=1;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
    t1.resize(n);for(int i=0;i<n;i++) t1[i]=(i<m1?a[i]:E(0,0));
    t2.resize(n);for(int i=0;i<n;i++) t2[i]=(i<m2?b[i]:E(0,0));
    FFT(t1,n,1),FFT(t2,n,1);for(int i=0;i<n;i++) t1[i]*=t2[i];
    FFT(t1,n,-1),c.resize(m1+m2-1);for(int i=0;i<m1+m2-1;i++) c[i]=t1[i];
    return 0;
}
vector<lint> clt;
int cc=0;
inline int cl(vector<E> &a)
{
    lint jw=0;int n=(int)a.size();clt.resize(n);
    for(int i=0;i<n;i++) clt[i]=(lint)(a[i].x+0.5);
    for(int i=0;i<n;i++) jw=(clt[i]+=jw)/BAS,clt[i]%=BAS;
    while(jw) clt.push_back(jw%BAS),jw/=BAS,n++;
    while(n>0&&!clt[n-1]) n--;if(!n) n=1,clt[0]=0;a.resize(n);
    for(int i=0;i<n;i++) a[i]=E(clt[i],0);return 0;
}
inline int solve(int l,int r)
{
    if(l==r) return 0;int mid=(l+r)>>1;solve(l,mid),solve(mid+1,r);
    tms(b[l],a[mid+1],a[mid+1]),tms(b[l],b[mid+1],b[l]);
    int L=(int)a[l].size(),R=(int)a[mid+1].size(),k=max(L,R);
    a[l].resize(k);for(int i=L;i<k;i++) a[l][i]=E(0,0);
    a[r].resize(k);for(int i=R;i<k;i++) a[mid+1][i]=E(0,0);
    for(int i=0;i<(int)a[l].size();i++) a[l][i]+=a[mid+1][i];
    cl(a[l]),cl(b[l]);
    return 0;
}
int main()
{
    int n=inn();
    for(int i=1;i<=n;i++) b[i].push_back(inn()),cl(b[i]);
    for(int i=1;i<=n;i++) a[i].push_back(inn()),cl(a[i]);
    return solve(1,n),show(a[1],0),0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值