BZOJ5302: [Haoi2018]奇怪的背包

BZOJ5302: [Haoi2018]奇怪的背包

https://lydsy.com/JudgeOnline/problem.php?id=5302

分析:

• 方程\(\sum\limits_{i=1}^nx_ia_i=y\)有整数解的条件是\(gcd|y\)。
• 对于这道题，我们可以直接把\(P\)当成一个可以提供负值 or 可以抵消负值的存在。
• 那么这道题的条件就是\(gcd(d,P)|q_i\)，其中\(d\)是选出那些数的\(gcd\)。
• 问题缩小到了\(P\)的约数这个范围，最多\(1400\)个。
• 然后一通预处理就做完了，询问直接用\(gcd(P,q_i)\)对应答案来回答。

代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
using namespace std;
#define N 1000050
#define mod 1000000007
typedef long long ll;
const int S=100000;
char buf[100000],*p1,*p2;
#define nc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++)
int rd() {
    int x=0; char s=nc();
    while(s<'0') s=nc();
    while(s>='0') x=(((x<<2)+x)<<1)+s-'0',s=nc();
    return x;
}
int n,m,a[N],P;
int id1[S+10],id2[S+10],w[1450],tot,b[N];
ll f[2][1450],ans[1450];
int gcd(int x,int y) {return y?gcd(y,x%y):x;}
inline void upd(ll &x,ll y) {x+=y; if(x>=mod)x-=mod;}
inline int pos(int x) {return x<=S?id1[x]:id2[P/x];}
inline ll qp(ll x,ll y) {
    ll re=1;
    for(;y;y>>=1,x=x*x%mod) if(y&1) re=re*x%mod; return re;
}
char pbuf[100000],*pp=pbuf;
void push(const char c) {
    if(pp-pbuf==100000) fwrite(pbuf,1,100000,stdout),pp=pbuf;
    *pp++=c; 
}
int sta[20],tp;
void write(int x) {
    do {
        sta[++tp]=x%10;
        x/=10;
    }while(x);
    while(tp) push(sta[tp--]+'0');
    push('\n');
}
int main() {
    n=rd(),m=rd(),P=rd();
    int i,j;
    for(i=1;i<=n;i++) a[i]=rd();
    for(i=1;i*i<=P;i++) {
        if(P%i==0) {
            w[++tot]=i;
            if(i*i!=P) w[++tot]=P/i;
        }
    }
    sort(w+1,w+tot+1);
    for(i=1;i<=tot;i++) {
        if(w[i]<=S) id1[w[i]]=i;
        else id2[P/w[i]]=i;
    }
    for(i=1;i<=n;i++) b[pos(gcd(a[i],P))]++;
    for(i=1;i<=tot;i++) b[i]=qp(2,b[i])-1;
    f[0][0]=1;
    for(i=0;i<tot;i++) {
        int i0=i&1,i1=(i+1)&1;
        for(j=0;j<=i+1;j++) if(f[i0][j]) {
            f[i1][j]=(f[i1][j]+f[i0][j])%mod;
            int k=pos(gcd(w[j],w[i+1]));
            f[i1][k]=(f[i1][k]+f[i0][j]*b[i+1])%mod;
        }
        memset(f[i0],0,sizeof(f[i0]));
    }
    int l=tot&1;
    for(i=1;i<=tot;i++) {
        for(j=1;j<=i;j++) if(w[i]%w[j]==0) {
            ans[i]=(ans[i]+f[l][j]);
        }
        ans[i]%=mod;
    }
    while(m--) {
        int x; x=rd();
        write(ans[pos(gcd(x,P))]);
    }
    fwrite(pbuf,1,pp-pbuf,stdout);
}

转载于:https://www.cnblogs.com/suika/p/10230203.html