牛客网-小马哥的超级盐水

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ACMER_2333/article/details/79874973

题目大意:给你n杯水,并且每杯水的含盐量和水量都已知,问你有多少种方案可以使混合后的含盐量为x/y。
分析:由于n的最大值为35,如果直接暴力深搜会超时,因此我们可以用一次折半查找来优化。由于水的含盐量最终为x/y,即可得到sum(a)/sum(b)==x/y,化简可得sum(a*y)-sum(b*x)=0,因此我们可以把n分成两部分,在(1,mid)的深搜的过程中用map(pre)来存储相应含盐量的方案数,在(mid+1,n)深搜中,当我们每次深搜结束时,我们就可以加上pre[-(a*y-b*x)]来统计总的方案数。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
int n,x,y;
int mid;
int a[1000000];
int b[1000000];
map<long,long> pre;
long long sum;
void dfs1(int a1,int b1,int num)
{
    if(num==mid+1)
    {
        pre[a1*y-b1*x]++;
        return ;
    }
    dfs1(a1+a[num],b1+b[num],++num);
    dfs1(a1,b1,num);
}
void dfs2(int a1,int b1,int num)
{
    if(num==n+1)
    {
        sum+=pre[b1*x-a1*y];
        return ;
    }
    dfs2(a1+a[num],b1+b[num],++num);
    dfs2(a1,b1,num);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        pre.clear();
        a[0]=0;
        b[0]=0;
        sum=0;
        scanf("%d%d%d",&n,&x,&y);
        for(int i=1;i<=n;i++)
            scanf("%d%d",&a[i],&b[i]);
        mid=n/2;
        dfs1(0,0,1);
        dfs2(0,0,mid+1);
        printf("%lld\n",sum-1);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/ACMER_2333/article/details/79874973