HDU 1709 (母函数)

/*

题目:给出一定(n)数量的砝码,每个砝码重a1,a2,a3。。。an,

问题:求出【1~sum】中(sum为各砝码总和)不能被称出的重量;

问题关键:天平两边都可以放砝码,

放在同一端:a[j+k]+=a[j]不同端a[abs(j-k)]+=a[j]

         假设原来的砝码都放在右端,则可以把新加的砝码放在左端,得到新重量,此时a[abs(j-k)]+=a[j]

http://acm.hdu.edu.cn/showproblem.php?pid=1709点击打开链接

*/



#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
#define maxn 101 


int a[maxn*maxn],b[maxn*maxn],c[maxn];
int main()
{
    int n;
    while(cin>>n)
    {
        int sum = 0;
        for(int i = 0; i < n; i++)
        {
            cin>>c[i];
            sum+=c[i];
        }
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i = 0; i <= c[0]; i+=c[0])
            a[i] = 1;
        //int now = c[0];
        for(int i = 1; i < n; i++)  //1。i代表第i个表达式 
        {
            for(int j = 0; j <= sum; j++)  //2。j表示每个表达式的第i项 
            {
                for(int k = 0; k <= c[i]; k+=c[i])  //k代表每个表达式第i项的指数 
                {
                    b[j+k] += a[j];
                    b[abs(j-k)] +=a[j];
                }
            }
            //now += c[i];
            for(int j = 0; j <= sum; j++)
            {
                a[j] = b[j];
                b[j] = 0;
            }
        }
        int num = 0;
        for(int i = 0; i <= sum; i++ )
        {
            if(!a[i])
                num++;
        }
        cout<<num<<endl;
        int p = 1;
        if(num){
            for(int i = 0; i <= sum; i++)
            {
                if(!a[i])
                {
                    if(!p)
                    cout<<" "<<i;
                    else
                    {
                        cout<<i;
                        p = 0;
                    }    
                }
                    
            }
            cout<<endl;
        }
        
    }
}


你可能感兴趣的:(HDU 1709 (母函数))