A1009. Product of Polynomials (25)
一般想法是,设置a, b两个数组,输入完之后再用两层循环去计算
#include <stdio.h>
int main(){
int K, i, exp;
double coef;
// 输入K
scanf("%d", &K);
// key代表指数,value代表系数
double a[1001] = {0.0}, b[1001] = {0.0}, res[2001] = {0.0}; // res 数组大小要为2001
// 输入数据
for(i = 0; i < K; i++){
scanf("%d %lf", &exp, &coef);
a[exp] += coef;
}
scanf("%d", &K);
for(i = 0; i < K; i++){
scanf("%d %lf", &exp, &coef);
b[exp] += coef;
}
// 计算系数
for(i = 0; i <= 1000; i++){
for(int j = 0; j <= 1000; j++){
res[i + j] += a[i] * b[j];
}
}
int count = 0;
for(i = 0; i <= 2000; i++){
if (res[i] != 0) {
count++;
}
}
// 输出结果
printf("%d", count);
for(i = 2000; i >= 0; i--){
if(res[i] != 0)
printf(" %d %.1lf", i, res[i]);
}
return 0;
}
更优化的办法就是不必声明第二个数组,在进行第二组输入的时候,就对数据进行处理
#include <stdio.h>
int main(){
int K, i, j, exp, count = 0;;
double coef;
// 输入K
scanf("%d", &K);
// key代表指数,value代表系数
double a[1001] = {0.0}, res[2001] = {0.0}; // res 数组大小要为2001
// 输入第一组数据
for(i = 0; i < K; i++){
scanf("%d %lf", &exp, &coef);
a[exp] += coef;
}
// 输入第二组数据,并进行运算
scanf("%d", &K);
for(i = 0; i < K; i++){
scanf("%d %lf", &exp, &coef);
// 和第一组数据中每一项相乘
for(j = 0; j <= 1000; j++)
res[exp + j] += coef * a[j];
}
for(int i = 0; i <= 2000; i++){
if (res[i] != 0)
count++;
}
// 输出结果
printf("%d", count);
for(i = 2000; i >= 0; i--){
if(res[i] != 0)
printf(" %d %.1lf", i, res[i]);
}
return 0;
}