洛谷P1164 小A点菜(母函数)
题目背景
uim神犇拿到了uoi的ra(镭牌)后,立刻拉着基友小A到了一家……餐馆,很低端的那种。uim指着墙上的价目表(太低级了没有菜单),说:“随便点”。
题目描述
不过uim由于买了一些辅(e)辅(ro)书,口袋里只剩M元(M<=10000)。餐馆虽低端,但是菜品种类不少,有N种(N<=100),第i种卖ai元(ai<=1000)。由于是很低端的餐馆,所以每种菜只有一份。
小A奉行“不把钱吃光不罢休”,所以他点单一定刚好吧uim身上所有钱花完。他想知道有多少种点菜方法。
由于小A肚子太饿,所以最多只能等待1秒。
输入输出格式
输入格式:
第一行是两个数字,表示N和M。第二行起N个正数ai(可以有相同的数字,每个数字均在1000以内)。
输出格式:
一个正整数,表示点菜方案数。输入输出样例
输入样例#1:
4 41 1 2 2
输出样例#1:
3解题分析
用母函数求解。由于每一种菜只有一份,因此第i种菜的母函数为1+x^ai,将所有菜的母函数相乘,求得指数为m的系数即可。
#include
#include
#include
#include
using namespace std;
int n, m;
struct node{
int e;
int f;
node *next;
node():next(NULL){
}
void set(int e1, int f1){
e = e1;
f = f1;
}
};
void add_item(node *now, int e, int f){
node *tmp;
while(now){
if(now->next == NULL){
tmp = new node;
tmp->set(e, f);
now->next = tmp;
return;
} else if(now->next->e == e){
now->next->f += f;
tmp = now->next;
if(tmp->f==0){
now->next = tmp->next;
delete tmp;
}
return;
} else if(now->next->e > e){
tmp = new node;
tmp->set(e, f);
tmp->next = now->next;
now->next = tmp;
return;
}
now = now->next;
}
}
void add_of_two(node *n1, node *n2){
node *tmp = n1, *tmp1 = n1->next, *tmp2 = n2->next, *tmp3;
while(tmp1!=NULL || tmp2!=NULL){
if(tmp1!=NULL && tmp2!=NULL && tmp1->e == tmp2->e){
tmp1->f += tmp2->f;
if(tmp1->f == 0){
tmp->next = tmp1->next;
delete tmp1;
tmp1 = tmp->next;
} else{
tmp = tmp1;
tmp1 = tmp1->next;
}
tmp2 = tmp2->next;
} else if(tmp2==NULL || tmp1!=NULL && tmp1->e < tmp2->e){
tmp = tmp1;
tmp1 = tmp1->next;
} else if(tmp1==NULL || tmp2!=NULL && tmp1->e > tmp2->e){
tmp3 = new node;
tmp3->set(tmp2->e, tmp2->f);
tmp3->next = tmp1;
tmp->next = tmp3;
tmp = tmp3;
tmp2 = tmp2->next;
}
}
}
node * mul_item(node *n1, int e, int f){
node *tmp = new node, *r = new node;
n1 = n1->next;
if(n1){
tmp->set(n1->e + e, n1->f * f);
tmp->next = NULL;
r->next = tmp;
}
n1 = n1->next;
while(n1){
tmp->next = new node;
tmp->next->set(n1->e + e, n1->f * f);
tmp->next->next = NULL;
tmp = tmp->next;
if(n1->e+e > m) // 优化
break;
n1=n1->next;
}
return r;
}
void get_i(int &x){
char ch = getchar();
x = 0;
while(!isdigit(ch)) ch = getchar();
while(isdigit(ch)){
x = x * 10 + ch - '0';
ch = getchar();
}
}
int main(){
ios::sync_with_stdio(false);
int i, p;
node *h1 = new node, *h2 = new node;
get_i(n), get_i(m);
for(i=0; i
next; while(h1){ if(h1->e == m){ cout<
f<
e>m){ cout<<0<
next; } return 0; }