PTA: 高精度运算 (20分)
题目:
分别给出N和A的值(都是整数,1<=N<=150, 0<=A<=15)。
请给出级数
∑
i
=
1
N
i
∗
A
i
\displaystyle \sum_{i=1}^N i*A^i
i=1∑Ni∗Ai的答案
输入格式:
有多组测试数据,每组测试数据占一行,该行包含2个整数N和A,以空格分隔。
输出格式:
对于输入每一行,在一行中输出相应于N和A的总和的整数值。
输入样例:
3 3
4 4
输出样例:
102
1252
附上我的解题代码,我的思路很简单,就是把每一项的和算出来进行累加。
这题还可以通过数学的等比等差数列推导n项和的思路来解,我数学底子不好,请自行尝试。
#include<bits/stdc++.h>
using namespace std;
//大数相加函数
string addStrings(string num1, string num2) {
string str;
int cur = 0, i = num1.size()-1, j = num2.size()-1;
while (i >= 0 || j >= 0 || cur != 0) {
if (i >= 0) cur += num1[i--] - '0';
if (j >= 0) cur += num2[j--] - '0';
str += to_string(cur % 10);
cur /= 10;
}
reverse(str.begin(), str.end());
return str;
}
//大数相乘函数
string multiply(string num1, string num2) {
int n1=num1.size();
int n2=num2.size();
string res(n1+n2,'0');
for(int i=n2-1;i>=0;i--){
for(int j=n1-1;j>=0;j--){
int temp=(res[i+j+1]-'0')+(num1[j]-'0')*(num2[i]-'0');
res[i+j+1]=temp%10+'0';//当前位
res[i+j]+=temp/10; //前一位加上进位,res[i+j]已经初始化为'0',加上int类型自动转化为char,所以此处不加'0'
}
}
//去除首位'0'
for(int i=0;i<n1+n2;i++){
if(res[i]!='0')
return res.substr(i);
}
return "0";
}
int main(){
int N,A;
string s[10000]; s[0] = '1';//保存每一次幂的值,s[0]=1, s[1]=A^1, s[2]=A^2.......s[N]=A^N
while (cin >> N >> A){
string n,a,sum; //n是N的字符串表示,a是A的字符串表示
int i = 1;
n = to_string(N);
a = to_string(A);
while(i<=N){
s[i] = multiply(a,s[i-1]); //每次计算都是上一次幂的值乘以A,
//因为上一次的值保存在s[i-1]中,减少了乘的次数
//如计算4^7 = s[6]*4,不用七次相乘,4*4*4*4*4*4*4 变成 s[6]*4
// 同时s[6]*4的值赋给了s[7].
sum = addStrings(sum,multiply(s[i],to_string(i)));//每次与i乘过后与之前的值加起来
i++;
}
cout<<sum<<endl;
}
return 0;
}
题目中的to_string()系列函数将数值转换成字符串形式。注意,不支持C++ 11标准的编译器可能不支持to_string()。
Dev-C++软件可以通过修改工具中的编译器选项,在编译时加入以下命令打上勾并在下方填入-std=c++11。如下图: