蓝桥杯校园选拔之位平方和

把一个整数的每个数位都平方后求和，又得到一个整数，我们称这个整数为：位平方和。
对新得到的整数仍然可以继续这一运算过程。
比如，给定整数为4，则一系列的运算结果为：
16,37,58,89,....
本题的要求是，已知一个整数x，求第n步的运算结果。
数据格式要求：
输入，两个整数x n，中间以空格分开。表示求x的第n步位平方和。其中，x，n都大于0，且小于100000。

输出，一个整数，表示所求结果。
例如，

输入：

4 3

则程序应该输出：

58
再例如，

输入：

1314 10

则程序应该输出：

20
资源约定：

峰值内存消耗 < 256M

CPU消耗  < 1000ms

#include<iostream>
using namespace std;
void Separation(int number, int bit, int separation[]);
int sumOfSquares(int number, int bit, int separation[]);
int main() {
int separation[100000];
int number, n,a,bit=0;
cin >> number >> n;
a = number;
while (n--) {
memset(separation, 0, sizeof(separation));
while (a) {
bit++;
a = a / 10;
}
Separation(number, bit, separation);
a=number=sumOfSquares(number, bit, separation);
}
cout << number<<endl;
system("pause");
return 0;
}
//分离
void Separation(int number,int bit,int separation[]) {
for (int i = 0; i < bit;i++) {
separation[i] = (int)(number / pow(10, i)) % 10;
}
}
//求位平方和
int sumOfSquares(int number,int bit,int separation[]) {
int sum=0;
for (int i = 0; i < bit;i++) {
sum += pow(separation[i],2);
}
return sum;
}