【洛谷】【模拟+栈】P4711 「化学」相对分子质量

【题目传送门：】 [戳] (https://www.luogu.org/problemnew/show/P4711)

[算法分析：]

关于一个分子拆分后的产物，一共有三种情况：

• 原子

• 原子团

• 水合物

关于原子的获取可以使用get_ele()函数，由于不是一位的元素就是两位的元素，只要看看这个大写字母的下一位是不是小写字母就行

返回获取的元素的string

关于系数的获取——使用函数get_num()——与快读的写法很接近，当没有系数的时候（处理后系数还是初始值没有被更新），系数是1

如何获取原子团：

在出现左括号的时候使用get_group()函数

使用get_ele()不停地获取括号内的元素直到下一位是系数或者不是元素，

当下一位是系数的时候获取系数，并且获取这一位的元素，将获取的系数与获取的元素相乘。

其他：

• 相对原子质量是实数，我们可以把他们都乘以10变成整数，有效地防止了被卡精度

• 在处理水合物的时候记得水的相对分子质量不应是18而是180

• 使用map来快速地获取每个元素的相对原子质量。

([Code:])

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;

const int MAXN = 1000 + 1;

map<string, int> m;

string str;
int e, c;
int ele[MAXN], coe[MAXN];

void init() {
	m["H"] = 10;
    m["C"] = 120;
    m["N"] = 140;
    m["O"] = 160;
    m["F"] = 190;
    m["Na"] = 230;
    m["Mg"] = 240;
    m["Al"] = 270;
    m["Si"] = 280;
    m["P"] = 310;
    m["S"] = 320;
    m["Cl"] = 355;
    m["K"] = 390;
    m["Ca"] = 400;
    m["Mn"] = 550;
    m["Fe"] = 560;
    m["Cu"] = 640;
    m["Zn"] = 650;
    m["Ag"] = 1080;
    m["I"] = 1270;
    m["Ba"] = 1370;
    m["Hf"] = 1785;
    m["Pt"] = 1950;
    m["Au"] = 1970;
    m["Hg"] = 2010;
}

int get_num(int k) {
	int i = k + 2, num = 0;
	while(str[i] >= '0' && str[i] <= '9')
		num = (num << 3) + (num << 1) + str[i] - '0', ++i;
	if(!num) num = 1;
	coe[++c] += num;
	return i;
}

inline string get_ele(int &k) {
	string ret;
	ret += str[k];
	if(str[k + 1] >= 'a' && str[k + 1] <= 'z')
		ret += str[++k];
	return ret;
}

int get_group(int k) {
	int i = k + 1, num = 0;
	while(str[i] >= 'A' && str[i] <= 'Z' && str[i+1] != '_')
		num += m[get_ele(i)], ++i;
	if(str[i + 1] == '_' && str[i] != ')') {
		i = get_num(i + 1);
		i -= 4;
		num += m[get_ele(i)] * coe[c];
		i += 4;
		coe[c--] = 0;
	}
	
	ele[++e] = num;
	return i;
}

int get_h2o(int k) {
	int i = k + 1, num = 0;
	while(str[i] >= '0' && str[i] <= '9')
		num = (num << 3) + (num << 1) + str[i] - '0', ++i;
	if(!num) num = 1;
	ele[++e] = 180, coe[++c] = num;
	return i + 5;
}

int main() {
	init();
	cin >> str;
	int l = str.size();
	for(int i=0; i<l; ++i) {
		if(str[i] == '_') {
			i = get_num(i);
			continue;
		}
		if(str[i] == '(') {
			i = get_group(i);
			continue;
		}
		if(str[i] == '~') {
			i = get_h2o(i);
			continue;
		}
		if(str[i] >= 'A' && str[i] <= 'Z') {
			ele[++e] = m[get_ele(i)];
			if(str[i + 1] != '_') coe[++c] = 1;
		}
	}
	int ans = 0;
	for(int i=1; i<=e; ++i)
		ans += ele[i] * coe[i];
	cout << ans / 10.0;
}