杠杆

【问题描述】
可以将一个杠杆描述为一个字符串，将字符串的长度表示为|s|，那么杠杆看起来很像具有一个支点的水平棒，可以看成是在OX轴上的点0到|s-1|之间的一段。
杠杆描述的解码如下：
如果字符串的第 i 个字符为“^”，意指在坐标 i 的位置水平棒下是支点
如果字符串的第 i 个字符为“=”，意指在坐标 i 的位置水平棒上面没有任何内容
如果字符串的第 i 个字符为c(1-9)，意指在坐标 i 的位置上的水平棒上有一个质量为c的重量
给出杠杆的描述，输出它平衡与否。忽略杠杆本身的重量，假设最初它是平衡的，所有的重量都同时施加在上面，杠杆要么向左倾斜，要么向右倾斜，要么处于平衡状态。
【输入形式】
输入的第一行为一个正整数T，表示有多组测试用例
接下来T行，每行对应一个测试用例，输入为一个非空字符串s(3 ≤|s| ≤106)，由数字1-9以及字符“^{”和“=”组成，输入保证这行有且只有一个“}”字符，输入保证字符“^”不在字符串的两端。
【输出形式】
输出有T行，每行对应一个测试用例，如果杠杆向左倾斜，输出“left”，向右倾斜，输出“right”，否则输出“balance”。
【样例输入】
2
9===^1
41^52
【样例输出】
left
balance
【样例说明】

解题思路：
先考虑一组数据，记下支点位置，分别求左边和右边的质量
质量=距离支点的距离*数字

#include <iostream>
#include <string>
#include <vector>
using namespace std;

//判断杠杆上是否为数字
bool number(char ch)
{
	if (ch >= 49 && ch <= 57)  return true;
	return false;
}

int lever(string str)
{
	int len = str.size(), fulcrum = 0;  //fulcrum为支点坐标
	while (str[fulcrum] != '^') fulcrum++;  //支点的位置
	int LeftWeight = 0, RightWeight = 0;
	for (int i = 0; i < len; i++)  //分别求左边和右边的重量
	{
		if(i < fulcrum && number(str[i])) 
		{
			int lnum = str[i] - '0', l = 0;
			l = fulcrum - i;   //左边数字离支点的位置
			LeftWeight += l * lnum;
		}
		if (i > fulcrum && number(str[i]))
		{
			int rnum = str[i] - '0', r = 0;
			r = i - fulcrum;   //右边数字离支点的位置
			RightWeight += r * rnum;
		}
	}
	if (LeftWeight > RightWeight)
	{
		return -1;
	}
	else if (LeftWeight == RightWeight)
	{
		return 0;
	}
	else
	{
		return 1;
	}
}

int main()
{
	int n;
	cin >> n;
	vector<string>  str(n);
	vector<int>  a(n);  //储存每个杠杆判断的结果
	for (int i = 0; i < n; i++)
	{
		cin >> str[i];
	}
	for (int i = 0; i < n; i++)
	{
		a[i] = lever(str[i]);
	}
	for (int i = 0; i < n; i++)
	{
		if (a[i] == -1) cout << "left" << endl;
		else if (a[i] == 0) cout << "balance" << endl;
		else cout << "right" << endl;
	}
}