N - DAG优化
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+C
Output
B=B+B
A=C+C
#include <iostream>
#include <vector>
using namespace std;
struct Node {
char id;
int left = -1, right = -1;
vector<char> vars;
} node[110];
char s[10];
char ans[110][10];
int flag[110];
int n, cnt;
bool find(int i, char ch) {
for (char c : node[i].vars) {
if (c == ch) return true;
}
return false;
}
int add_node(char ch) {
for (int i = 0; i < cnt; i++) {
if (node[i].id == ch || find(i, ch)) return i;
}
node[cnt].id = ch;
return cnt++;
}
void add_op(char var, char op, int l, int r) {
for (int i = 0; i < cnt; i++) {
if (node[i].id == op && node[i].left == l && node[i].right == r) {
node[i].vars.push_back(var);
return;
}
}
node[cnt].id = op;
node[cnt].left = l;
node[cnt].right = r;
node[cnt].vars.push_back(var);
cnt++;
}
char exist(int i) {
char ans = 0;
for (char c : node[i].vars) {
if (c == 'A' || c == 'B') ans = c;
}
return ans == 0 ? node[i].vars[0] : ans;
}
void dfs(int x) {
if (node[x].left != -1) {
flag[x] = 1;
dfs(node[x].left);
dfs(node[x].right);
}
}
int main() {
cin >> n;
for (int i = 0; i < n; i++) {
cin >> s;
int l = add_node(s[2]);
int r = add_node(s[4]);
add_op(s[0], s[3], l, r);
}
for (int i = 0; i < cnt; i++) {
if (node[i].left != -1) {
ans[i][0] = exist(i);
ans[i][1] = '=';
Node ll = node[node[i].left];
Node rr = node[node[i].right];
ans[i][2] = ll.vars.size() > 0 ? exist(node[i].left) : ll.id;
ans[i][3] = node[i].id;
ans[i][4] = rr.vars.size() > 0 ? exist(node[i].right) : rr.id;
ans[i][5] = 0;
}
}
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;
}