练题——磁盘容量排序

题目描述：
磁盘的容量单位常用的有M，G，T这三个等级，它们之间的换算关系为1T = 1024G，1G =
1024M，现在给定n块磁盘的容量，请对它们按从小到大的顺序进行稳定排序，例如给定5
块盘的容量，1T，20M，3G，10G6T，3M12G9M排序后的结果为20M，3G，3M12G9M，
1T，10G6T。注意单位可以重复出现，上述3M12G9M表示的容量即为3M+12G+9M，和
12M12G相等。
输入描述：
输入第一行包含一个整数n(2 <= n <= 100)，表示磁盘的个数，接下的n行，每行一个字符串
(长度大于2，小于30)，表示磁盘的容量，由一个或多个格式为mv的子串组成，其中m表示
容量大小，v表示容量单位，例如20M，1T，30G，10G6T，3M12G9M。
磁盘容量m的范围为1到1024的正整数，容量单位v的范围只包含题目中提到的M，G，T三
种，换算关系如题目描述。
输出描述：
输出n行，表示n块磁盘容量排序后的结果。

#include <string>
#include <iostream>
#include <vector>
#include <stdlib.h>

std::vector<std::string> GetStr(std::string str)
{
	std::vector<std::string> tgm;
	std::string s;
	for (int i = 0; i < str.size(); i++)
	{
		if (str.at(i) % 'M' == 0 || str.at(i) % 'G' == 0 || str.at(i) % 'T' == 0)
		{
			if (str.at(i) % 'M' == 0)
			{
				s = s + 'M';
			}

			else if (str.at(i) % 'G' == 0)
			{
				s = s + 'G';
			}

			else if (str.at(i) % 'T' == 0)
			{
				s = s + 'T';
			}

			tgm.push_back(s);
			s.clear();
		}

		else
		{
			s = s + str.at(i);
		}
	}

	return tgm;
}

int Getnum(std::string str)
{
	return atoi(str.substr(0, str.length() - 1).c_str());
}

std::vector<int> merge(std::vector<std::string> strs)
{
	std::vector<int> v = { 0,0,0 };
	for (int i = 0; i < strs.size(); i++)
	{
		if (strs.at(i).at(strs.at(i).size() - 1) == 'T')
		{
			v.at(0) = Getnum(strs.at(i));
		}

		else if (strs.at(i).at(strs.at(i).size() - 1) == 'G')
		{
			v.at(1) = Getnum(strs.at(i));
		}

		else if (strs.at(i).at(strs.at(i).size() - 1) == 'M')
		{
			v.at(2) = Getnum(strs.at(i));
		}
	}

	if (v.at(2) >= 1024)
	{
		v.at(2) = v.at(2) % 1024;
		v.at(1) = v.at(1) + v.at(2) / 1024;
	}

	if (v.at(1) >= 1024)
	{
		v.at(1) = v.at(1) % 1024;
		v.at(0) = v.at(0) + v.at(1) / 1024;
	}

	return v;
}

int Compare(std::string str1, std::string str2)
{
	std::vector<std::string> strs1 = GetStr(str1);
	std::vector<std::string> strs2 = GetStr(str2);
	std::vector<int> num1 = merge(strs1);
	std::vector<int> num2 = merge(strs2);

	if (num1.at(0) > num2.at(0))
	{
		return 1;
	}

	else if (num1.at(0) == num2.at(0))
	{
		if (num1.at(1) > num2.at(1))
		{
			return 1;
		}

		else if (num1.at(1) == num2.at(1))
		{
			if (num1.at(2) > num2.at(2))
			{
				return 1;
			}

			else if (num1.at(2) == num2.at(2))
			{
				return 0;
			}

			else
			{
				return -1;
			}
		}

		else
		{
			return -1;
		}
	}

	else
	{
		return -1;
	}
}

bool isneed(std::vector<std::string> strs)
{
	for (int i = 0; i < strs.size() - 1; i++)
	{
		if (Compare(strs.at(i), strs.at(i + 1)) == 1)
		{
			return true;
		}
	}

	return false;
}

void sort(std::vector<std::string> &strs)
{
	while (isneed(strs))
	{
		for (int i = 0; i < strs.size() - 1; i++)
		{
			if (Compare(strs.at(i), strs.at(i + 1)) == 1)
			{
				std::string s = strs.at(i);
				strs.at(i) = strs.at(i + 1);
				strs.at(i + 1) = s;
			}
		}
	}
}

int main()
{
	//输入
	int N;
	std::cin >> N;
	std::vector<std::string> str(N);
	for (int i = 0; i<str.size(); i++)
	{
		std::cin >> str.at(i);
	}

	sort(str);
	for (int i = 0; i<str.size(); i++)
	{
		std::cout << str.at(i) << std::endl;
	}

	return 0;
}