一本通 P1681 统计方案

Posted 2020-11-22 wangjunrui

题目链接

这题其实思路很简单，先构造前16位的各种余数的方法，最多也(2^{16})种，然后再次从后面开始再搜索统计一遍，就可以了，AC代码如下：

#include<cstdio>
#include<map>
#define re register
using namespace std;
template<typename T>
inline void read(T&x)
{
    x=0;
    char s=getchar();
    bool f=false;
    while(!(s>='0'&&s<='9'))
    {
        if(s=='-')
            f=true;
        s=getchar();
    }
    while(s>='0'&&s<='9')
    {
        x=(x<<1)+(x<<3)+s-'0';
        s=getchar();
    }
    if(f)
        x=(~x)+1;
}
const int N=(1<<16)+10;
map<int,int>cnt;
int p,c,n,a[10010];
const int mod=1e9+7;
long long ans;
inline long long quickpow(long long a,int b)
{
    long long res=1ll;
    while(b)
    {
        if(b&1)
            res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}
inline void search1(int t,long long yu)
{
    if(t==(n>>1)+1)
    {
        cnt[yu]++;
        return;
    }
    search1(t+1,yu);
    search1(t+1,(a[t]*yu)%p);
}
inline void search2(int t,long long yu)
{
    if(t>n)
    {
        ans=(ans+cnt[yu])%mod;
        return;
    }
    long long temp=yu*(quickpow(a[t]%p,p-2))%p;
    //ans=(ans+cnt[temp])% mod;
    search2(t+1,temp);
    search2(t+1,yu);
}
int main()
{
    read(n);
    read(p);
    read(c);
    if(c>=p)
    {
        printf("0
");
        return 0;
    }
    for(re int i=1; i<=n; i++)
        read(a[i]);
    search1(1,1);
    //cnt[1]++;
    //printf("%d
",cnt[1]);
    search2((n>>1)+1,c);
    if(c==1)
        ans--;
    printf("%lld
",ans);
    return 0;
}