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

把一个整数的每个数位都平方后求和，又得到一个整数，我们称这个整数为：位平方和。

对新得到的整数仍然可以继续这一运算过程。

比如，给定整数为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;
}