[DFS][hash]方程的解数

最新推荐文章于 2020-07-27 10:39:57 发布

Vagari_

最新推荐文章于 2020-07-27 10:39:57 发布

阅读量385

点赞数

本文链接：https://blog.csdn.net/SSL_QYH0Ice/article/details/78757237

版权

该博客探讨了一道关于求解方程整数解数目的问题。当数据范围中m达到150，直接枚举会导致时间复杂度过高。通过观察发现150^3小于1000万且是最大范围的一半，博主提出将问题转化为两部分：枚举150^3的前三个数并计算其和，然后枚举后三个数并取相反数，使用哈希来查找是否存在对应的解。总时间复杂度为O(2*150^3)。

摘要由CSDN通过智能技术生成

题目描述
这里写图片描述
Input

　　第1行包含一个整数n。第2行包含一个整数M。第3行到第n+2行，每行包含两个整数，分别表示ki和pi。两个整数之间用一个空格隔开。第3行的数据对应i=1，第n+2行的数据对应i=n。

Output

仅一行，包含一个整数，表示方程的整数解的个数。

数据范围
1≤n≤6
1≤m≤150
所有值的绝对值加起来不超过2^31-1

分析
首先我们看到数据范围
m为150，那么这题可能要枚举了
但是n为6，150^6绝壁爆炸的，我们要想办法简化时间。
这时候我们看向150^3这个数，它小于1000万，而且刚好是最大范围的一半，肯定有什么意义。
仔细想一想，如果前面半边的数为S，那么后面得到-S就可以得到0了
这题就转换为150^3的枚举前三个数的可能总值，然后再枚举一次150^3后三个数，把它转为相反数以后哈希寻找就行啦
总时间复杂度：O(2*150^3)

#include <iostream>
#include <cstdio>
#include <cmath>
using namespace std;
int n,m,k[7],p[7],h[4000037][2],i,j,q=4000037;
long long mc[151][31],ans;
long long loc(long long x)
{
    long long i=x;
    if (i<0) i=-i;
    i%=q;
    while (h[i%q][0]!=0&&h[i%q][0]!=x)
    i++;
    return i%q;
}
void ins(long long x)
{
    long long i=loc(x);
    h[i][0]=x;
    h[i][1]++;
}
void mem(long long x)
{
    long long i=loc(x);
    if (h[i][0]==x)
    ans+=h[i][1];
}
void dqz(int x,long long s)
{
    int i;
    if (x==n/2)
    {
        for (i=1;i<=m;i++)
        ins(s+k[x]*mc[i][p[x]]);
    }
    else
    for (i=1;i<=m;i++)
    dqz(x+1,s+k[x]*mc[i][p[x]]);
}
void dzz(int x,long long s)
{
    int i;
    if (x==n)
    {
        for (i=1;i<=m;i++)
        mem(-1*(s+k[x]*mc[i][p[x]]));
    }
    else
    for (i=1;i<=m;i++)
    dzz(x+1,s+k[x]*mc[i][p[x]]);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
    scanf("%d%d",&k[i],&p[i]);
    if (n==1)
    {
        if (k[1]==0) printf("%d",m);
        else printf("0");
        return 0;
    }
    for (i=1;i<=30;i++)
    mc[1][i]=1;
    for (i=2;i<=m;i++)

    {
        mc[i][1]=i;
        for (j=2;j<=log(2147483647)/log(i);j++)
        mc[i][j]=mc[i][j-1]*i;
    }
    dqz(1,0);
    dzz(n/2+1,0);
    printf("%lld",ans);
}