找單詞問題
阿新 • • 發佈:2018-12-09
問題描述
Problem 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
解題思路:母函式問題;用指數代表價值,價值又為陣列的下標;用係數代表組成該價值的方案數,方案數為陣列中存的值;
程式碼如下
#include<stdio.h> #include<iostream> #include<cstring> using namespace std; int maxn = 50; int main(){ int num[maxn + 1]; int c1[maxn + 1]; int c2[maxn + 1]; int i; int j,k; int T; scanf("%d",&T); while(T --){ for(i = 1;i <= 26;i ++) scanf("%d", &num[i]); memset(c1, 0, sizeof c1); memset(c2, 0, sizeof c2); for(i = 0;i <= num[1];i++) c1[i] = 1; for(i = 2;i <= 26;i ++){ if(num[i] == 0) continue; for(j = 0;j <= maxn;j ++){ for(k = 0;k <= num[i] && j + k*i <= maxn;k ++) c2[j + k*i] += c1[j]; } for(j = 0;j <= maxn;j ++){ c1[j] = c2[j]; c2[j] = 0; } } int sum = 0; for(j = 1;j <= maxn;j ++) sum += c1[j]; printf("%d\n",sum); } return 0; }