相对分子质量

题目描述
小明最近迷上了化学，几乎天天在实验室做实验，但是很多实验生成的化学产物的相对分子质量令他很困惑，不知如何计算，请你编程帮他计算。
输入
输入的第一行是一个正整数n，表示有n组测试数据。
接下来n行每行输入一个字符串，表示某个分子式，分子式中只包含大写字母和数字。
注意：每个分子式长度没超过20.输入数据只包含8种元素，而这8种元素的相对原子质量如下：
H(1)，C(12)，N(14)，O(16)，F(19)，P(31)，S(32)，K(39)。
输出
对于每组输入，输出相对分子质量。
样例输入
4
H2O
KOH
CH4
SO2
样例输出
18
56
16
64

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std; 
int fun(char c)
{
	switch(c)
	{
		case 'H':return 1;break;
		case 'C':return 12;break;
		case 'N':return 14;break;
		case 'O':return 16;break;
		case 'F':return 19;break;
		case 'P':return 31;break;
		case 'S':return 32;break;
		case 'K':return 39;break;
		default:break;
	}
}
int main()
{
	int n,i,k,j;
	scanf("%d",&n);
	while(n--)
	{
		string s;
		cin>>s;
		int len=s.length();
		int sum=0;
		for(i=0;i<len;i++)
		{
			if(s[i+1]<='9'&&s[i+1]>'0')
			{
				int a=0;
				for(k=i+1;k<len;k++)    //计算字母后面的数字 
				{
					if(s[k]<='9'&&s[k]>'0')
					a=a*10+s[k]-'0';
					else break;
				}
				sum+=fun(s[i])*a;
			}
			else if(s[i]<='9'&&s[i]>='0') continue;
				else if(s[i]<'Z'&&s[i]>'A'&&(s[i+1]<'Z'&&s[i+1]>'A'||s[i+1]=='\0'))
				sum+=fun(s[i]);
		}
		cout<<sum<<endl;
	}
}

这道题不难，第一次做的时候是我想的太简单了，太懒了。。。思路是正确的，就是没想起来结尾的时候怎么做