HDOJ 2082 找单词

HDACM2082

此题采用母函数的方法，母函数就是用
f(x)=a0x^0+a1*x^1+a2*x^2+a3*x^3+…+an*x^n
用a0,a1,a2,a3…an表示值为x^0,x^1,x^2,x^3,…,x^n的个数。
举个例子 假如输入的是 1 1 1 0 0…… 0 0说明输入的是
求1个a、一个b和一个c组合的所有可能
单独由a组合有两种可能，0个a或1个a，则单独由a组合的价值有两种 0 和 1
可以用（x+y）表示
由a和b组合有四种可能，0个a+0个b、0个a+1个b、1个a+0个b和1个a+1个b
可以用(x+y)*(x+z)=(x*x+x*y+x*z+y*z)
由a、b和c组合有8中可能
（x*x+x*y+x*z+y*z）*（x+h）=x*x*x+x*x*y+…+y*z*h
a0(1)x*x*x 表示和值为0的组合的个数，a1(1)x*x*y表示和值为1的组合的个数… 所有价值低于50的组合的总数=1+1+1+1+1+1+1+1，但是题目要求至少要一个字母组成一个单词所有要价值为0的组合次数就不要加上去，最后的总数=a1+a2+…+a7=1+1+1+1+1+1+1=7
母函数原理链接

import java.util.Scanner;

public class Main{

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        while(n-->0){
            int[] m = new int[26];
            int[] value = new int[51];//用数组下标来表示字母组合的价值，用值来表示价值相同的不同组合个数
            int[] temp = new int[51];//用来做临时存储
            for (int i = 0; i < m.length; i++) {
                m[i] = sc.nextInt();
            }
            for (int i = 0; i <= m[0]; i++) {
                value[i] = 1; //先把字母a所组合的值的所有可能记录起来
            }   
            for (int i = 1; i < 26; i++) {
                for (int j = 0; j <= 50; j++) {
                    for (int k = 0; k <= m[i]&&(i+1)*k+j<=50; k++) {//(i+1)*k+j表示字母组合的值
                        temp[(i+1)*k+j] += value[j];
                    }
                }
                for (int j = 0; j <= 50; j++) {
                    value[j] = temp[j];//把最新的组合的总数代替原来的组合的总数
                    temp[j]=0;
                }
            }
            int sum = 0;
            for (int i = 1; i < value.length; i++) {
                sum += value[i]; //计算出价值为1~50的所有组合的总和
            }
            System.out.println(sum);
        }
    }
}