题目:
DAG优化
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
大家都学过了代码优化,其中有一个DAG优化,这次我们就练习这个操作。
Input
输入第一行为一个整数n(n < 100),表示该组输入的表达式的个数
之后n行为表达式,每个变量为一个字母,表达式仅包括二元运算 + - * /
例如:A=B+C
Output
通过构造DAG图,进行代码优化,只需要保留AB,删除无用变量,删除变量时,尽量保留最早出现的变量。
PS:保证AB的值不同
Sample Input
3 A=B+C B=B+B A=C+CSample Output
B=B+B A=C+C
解题代码:
#include <cstdio>
#include <vector>
using namespace std;
int n;
int cnt;
struct node
{
char id;
int left = -1, right = -1;
vector<char> var;
}node[110];
bool Find_var(int i, char c)
{
for (char j : node[i].var)
if (j == c) return 1;
return 0;
}
int Add_node(char c)
{
for (int i = cnt - 1; i >= 0; i--)
if (node[i].id == c || Find_var(i, c))
return i;
node[cnt].id = c;
return cnt++;
}
void Add_operator(char c, char op, int L, int R)
{
for (int i = cnt - 1; i >= 0; i--)
{
if (node[i].left == L && node[i].right == R && node[i].id == op)
{
node[i].var.push_back(c);
return;
}
}
node[cnt].id = op;
node[cnt].var.push_back(c);
node[cnt].left = L;
node[cnt].right = R;
cnt++;
}
char s[10];
char ans[110][10];
bool flag[110];
void DFS(int x)
{
if (node[x].left != -1)
{
flag[x] = 1;
DFS(node[x].left);
DFS(node[x].right);
}
}
int main()
{
cnt = 0;
scanf("%d", &n);
for (int i = 0; i < n; i++)
{
scanf("%s", s);
int L = Add_node(s[2]);
int R = Add_node(s[4]);
Add_operator(s[0], s[3], L, R);
}
for (int i = 0; i < cnt; i++)
{
if (node[i].left != -1)
{
ans[i][0] = node[i].var[0];
ans[i][1] = '=';
struct node LL = node[node[i].left], RR = node[node[i].right];
ans[i][2] = LL.var.size() > 0 ? LL.var[0]:LL.id;
ans[i][3] = node[i].id;
ans[i][4] = RR.var.size() > 0 ? RR.var[0] : RR.id;}
}
for (int i = cnt - 1; i >= 0; i--)
{
if (ans[i][0] == 'A')
{
DFS(i);
break;
}
}
for (int i = cnt - 1; i >= 0; i--)
{
if (ans[i][0] == 'B')
{
DFS(i);
break;
}
}
for (int i = 0; i < cnt; i++)
if (flag[i]) puts(ans[i]);
return 0;
}