【母函数】HDU-1171 Big Event in HDU

注解

1、母函数模板应用。先初始化最小的不为0个的单价，然后往后找每个存在的价格，运用母函数处理，求出每个价格对应的个数。
2、输入的是单价和对应的个数。要输出的是尽可能接近的两个能组成的价格。

代码

#include <iostream>
#include <cstring>
#include <cmath>

using namespace std;

const int MAX = 127501; //本题要找的是最大价值<=127500的（1、2、...、50均为100个），因此最大数组长度设为127501 
const int MAXNUM = 51; 

int a[MAX]; //保存每一轮结束后的结果
int b[MAX]; //保存当前轮对应的系数
int c[MAX]; //保存当前轮的结果

void init(int m, int n, int s[]) { //m:面额/价值，n:个数 s:系数数组 eg:价值为2的有3个，m=2,n=3
    for(int j=0; j<MAX; j++) {
        if(j%m==0 && j<=m*n) {
            s[j] = 1;
        } else {
            s[j] = 0;
        }
    }
}

void process() { //模拟母函数的手动处理过程
    memcpy(c, a, sizeof(a));
    for(int i=1; i<MAX; i++) {
        if(b[i]) {
            for(int j=0; j<MAX; j++) {
                if(i+j<MAX) {
                    c[i+j] += a[j];
                }
            }
        }
    }
    memcpy(a, c, sizeof(c));
}

int main() {

    int n;
    cin>>n;
    while(n>=0){
        
        memset(a, 0, sizeof(a));
        memset(b, 0, sizeof(b));
        memset(c, 0, sizeof(c));
        
        //初始化facility数组，facility[i]表示value为i的facility数目 
        int facility[MAXNUM];
        memset(facility, 0, sizeof(facility));
        int sum = 0;
        for(int i=0; i<n; i++) {
            int tmpa, tmpb;
            cin>>tmpa>>tmpb;
            facility[tmpa] = tmpb;
            sum += tmpa*tmpb;
        }
        
        //找到第一个value不为0的facility，初始化
        int pos = -1;
        for(int i=0; i<MAXNUM; i++) {
            if(facility[i]) {
                pos = i;
                break;
            }
        }
        init(pos, facility[pos], a);
        
        //找到其余value不为0的facility，计算出对应系数
        for(int i=pos+1; i<=MAXNUM; i++) {
            if(facility[i]) {
                init(i, facility[i], b);
                process();
            }
        }
        
        int ans = sum;
        for(int i=sum/2; i<sum; i++){
            if(a[i]){
                ans = i;
                break;
            }
        }
        cout<<max(ans, sum-ans)<<" "<<min(ans, sum-ans)<<endl;
        
        cin>>n; 
    }

    return 0;
}

结果