ACM模板-多项式朴素乘法

一、模板

int max_term = 55;// 最高次项
// a为n次多项式，b为m次多项式，结果放在c中，返回c多项式的最高次项，但是不会超过max_term
int polyMul(double * a, double *b, double *c, int n, int m) {
    for (int i = 0; i <= n; ++i) {
        for (int j = 0; j <= m; ++j) {
            if (i + j > max_term) break;
            else c[i + j] += a[i] * b[j];
        }
    }
    return m + n > max_term ? max_term : m + n;
}

二、例题

Description

假设有x1个字母A， x2个字母B,..... x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,..... 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词呢？单词的价值就是组成一个单词的所有字母的价值之和，比如，单词ACM的价值是1+3+14=18，单词HDU的价值是8+4+21=33。(组成的单词与排列顺序无关，比如ACM与CMA认为是同一个单词）。

Input

输入首先是一个整数N，代表测试实例的个数。
然后包括N行数据，每行包括26个<=20的整数x1,x2,.....x26.

Output

对于每个测试实例，请输出能找到的总价值<=50的单词数,每个实例的输出占一行。

Sample Input

2
1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
9 2 6 2 10 2 2 5 6 1 0 2 7 0 2 2 7 5 10 6 10 2 10 6 1 9

Sample Output

7
379297

Solution

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#define SIZE 60
using namespace std;
double A[SIZE], B[SIZE], C[SIZE];
int max_term = 55;// 最高次项
// a为n次多项式，b为m次多项式，结果放在c中，返回c多项式的最高次项，但是不会超过max_term
int polyMul(double * a, double *b, double *c, int n, int m) {
    for (int i = 0; i <= n; ++i) {
        for (int j = 0; j <= m; ++j) {
            if (i + j > max_term) break;
            else c[i + j] += a[i] * b[j];
        }
    }
    return m + n > max_term ? max_term : m + n;
}
int main()
{
    int n, num, P = 0;
    scanf("%d", &n);
    while (n--) {
        memset(A, 0, sizeof(A));
        long long int ret = 0;
        for (int i = 0; i < 26; ++i) {
            memset(B, 0, sizeof(B));
            memset(C, 0, sizeof(C));
            scanf("%d", &num);
            if (i == 0) A[0] = 1, P = 0;
            if (num == 0) continue;
            int k = 0;
            for (int j = 0; j <= num&&k<SIZE; ++j) {
                B[k] = 1.0;
                k += i + 1;
            }
            P = polyMul(A, B, C, P, k);
            memcpy(A, C, sizeof(A));
        }
        for (int i = 1; i <= 50; ++i)ret += A[i];
        cout << ret << endl;
    }
    return 0;
}