HDU 1082マトリクス乗算回数計算を半日書いて達成感がありますね

1840 ワード

この考えはもともと難しくなく、実現するのは少し複雑で、間違いやすい.デバッグが完了しました.達成感がある.、
#include <iostream>
#include <stack>
#include <vector>
#include <algorithm>
using namespace std;
int row[26], col[26];
char str[10000];
int count(int *a, int n)
{
	int flag = 0;
	if(n == 2)
		return 0;
	for(int i = 1 ; i < n-1; i += 2)
	{
		if(a[i] != a[i+1])
		{
			flag = 1;
			break;
		}
	}
	if(flag == 1)
		return -1;
	int mul = a[0];
	for(int i = 1; i < n; i += 2 )
	{
		mul *= a[i];
	}
	return mul;
}
int main()
{
	int t; 
	cin >> t;
	char temp;
	for(int i = 0 ;i < t ; i++)
	{
		getchar(); //  
		scanf("%c", &temp);  
		scanf("%d%d", &row[temp-'A'], &col[temp-'A']);
	}
	getchar();
	while(gets(str))
	{
		int n = strlen(str);
		if(n == 1)
		{
			cout << "0" << endl;
			continue;
		}

		stack<char> stk;
		int sum = 0;
		int flag = 0;
		for(int i = 0; i < n; i++)
		{
			if(str[i] != ')' )
			{
				if(isalpha(str[i]))
				{
					stk.push((char)row[str[i] - 'A']);
					stk.push((char)col[str[i] - 'A']);
				}
				if(str[i] == '(')
					stk.push('(');
			}
			if(str[i] == ')' || i == n-1)
			{
				int cnt = 0;
				int a[10000];
				while(!stk.empty())
				{
					if(stk.top() == '(')
					{
						stk.pop();
						break;
					}
					a[cnt++] = (int)stk.top();
					stk.pop();
				}
				reverse(a, a+ cnt);
				if(!stk.empty())
				{
					stk.push(a[0]);
					stk.push(a[cnt-1]);
				}
				int mul  = 1;
				mul = count(a, cnt);
				if(mul == -1)
				{
					flag = 1;
					break;
				}
				else
					sum += mul;
			}
			if(flag == 1)
				break;
		}
		if(flag == 1)
			cout << "error" << endl ;
		else
			cout << sum << endl;
	}
	return 0;
}