7-3 DAG图优化-A (15 分)(更新版)

7-3 DAG图优化-A (15 分)
大家都学过了代码优化，其中有一个DAG优化，这次我们就练习这个操作。

输入格式:
输入第一行为一个整数ｎ(n < 100)，表示该组输入的表达式的个数。

之后ｎ行为表达式，每个变量为一个字母，表达式仅包括二元运算 + - * / 。

例如：A=B+C 。

输出格式:
通过构造DAG图，进行代码优化，只需要保留AB，删除无用变量，删除变量时，尽量保留最早出现的变量。

PS:保证AB的值不同

输入样例:
3
A=B+C
B=B+B
A=C+C
输出样例:
B=B+B
A=C+C

#include <bits/stdc++.h>
using namespace std;
int cnt,n;               ///建立的第几条表达式
char s[10],ans[101][101];///ans存放所有的表达式。
bool flas[101];         ///存放优化后的表达式输出。
struct st
{
    char id;           ///每个结点的值。
    int left = -1;     ///左右子树的位置
    int right = -1;
    vector<char>var;   ///存放每个表达式的根。
}node[101];            ///存放表达式
bool find_var(int i,char c)  ///看是否已经建立过该结点
{
    int len = node[i].var.size();
    for(int k = 0; k < len; k++)
    {
        if(node[i].var[k] == c)
            return true;
    }
    return false;
}
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;  ///建立该结点，此结点为暂时的根结点，若其没有左子树后期会把他砍掉。（4），删除无用变量。
    return cnt++;
}
void add_operator(char c,char op,int l,int r)
{
    for(int i = cnt-1; i >= 0; i--) ///if is这种情况：A=B+C,D=B+C;优化类型一：删除多余运算。
    {
        if(op == node[i].id&&node[i].left == l&&node[i].right==r)
           {
            node[i].var.push_back(c);
            return ;
           }
    }///else增加整个表达式
    node[cnt].id=op;
    node[cnt].left=l;
    node[cnt].right=r;
    node[cnt].var.push_back(c);
    cnt++;
}
void DFS(int i)
{
    if(node[i].left!=-1)
    {
        flas[i] = 1;
        DFS(node[i].left);
        DFS(node[i].right);
    }
}
int main()
{
cnt = 0;
cin>>n;
for(int i = 0; i < n; i++)
{
    cin>>s;
    int l = add_node(s[2]);  ///（1）同时合并已知量
    int r = add_node(s[4]);
    add_operator(s[0],s[3],l,r); ///将输入的所有表达式copy给结构体数组node[i]（2），删除多余运算
}
for(int i = 0; i < cnt; i++)///遍历结构体数组node[i],把含有左子树的表达式整体copy给ans[]
{
    if(node[i].left!=-1)    ///只有该根节点存在左子树时才有用，没有就直接砍掉。
    {
        ans[i][0]=node[i].var[0];///保存根
        ans[i][1]='=';
        st ll = node[node[i].left];
        st rr = node[node[i].right];
        ans[i][2] = ll.var.size()>0?ll.var[0]:ll.id;///（3），搞定复写传播
        ans[i][3] = node[i].id;
        ans[i][4] = rr.var.size()>0?rr.var[0]: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(flas[i])
    {
        puts(ans[i]);
    }
}
    return 0;
}