数字之和

Problem Link：点击打开链接

题目描述

对于给定的正整数 n，计算其十进制形式下所有位置数字之和，并计算其平方的各位数字之和。 

输入描述:

每行输入数据包括一个正整数n(0<n<40000)，如果n=0 表示输入结束，并不用计算。

输出描述:

对于每个输入数据，计算其各位数字之和，以及其平方值的数字之和，输出在一行中，之间用一个空格分隔，但行末不要有空格。

输入例子:

4
12
97
39999
0

输出例子:

4 7
3 9
16 22
39 36

AC code：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<map>
#include<math.h>
#include<string.h>
#include<queue>
#include<vector>
#include<set>
#define LL long long
#define exp 1e-9
#define MAXN 1000010

using namespace std;

int sum(LL n)
{
	int s=0;
	while(n)
	{
		s+=(n%10);
		n/=10;
	}
	return s;
}

int main()
{
//	freopen("D:\\in.txt","r",stdin);
	int i,n,m,mm;
	while(scanf("%d",&n)!=EOF)
	{
		if(n==0) break;
		m=sum(n);
		mm=sum(n*n);
		printf("%d %d\n",m,mm);
	}
	return 0;
}