3500N

Description

大家都学过了代码优化，其中有一个DAG优化，这次我们就练习这个操作。

Input

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

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

例如：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 <string.h>
#include <vector>

using namespace std;

struct Node
{
    char id;
    int left=-1, right=-1;
    vector <char> ve;
}node[105];
int book[105], top;
char re[105][15];

int fin(int x, char c)
{
    if(node[x].id==c) return 1;
    int i, len;
    len = node[x].ve.size();
    for(i=0;i<len;i++)
    {
        if(node[x].ve[i]==c) break;
    }
    if(i<len) return 1;
    else return 0;
}

int add_node(char x)
{
    int i;
    for(i=0;i<top;i++)
    {
        if(fin(i, x)) break;
    }
    if(i<top) return i;
    else
    {
        Node nod;
        nod.id = x;
        node[top++] = nod;
        return top-1;
    }
}

void add_op(char re, char op, int left, int right)
{
    int i;
    for(i=0;i<top;i++)
    {
        if(node[i].id==op&&node[i].left==left&&node[i].right==right) break;
    }
    if(i<top) node[i].ve.push_back(re);
    else
    {
        Node nod;
        nod.id = op;
        nod.left = left;
        nod.right = right;
        nod.ve.push_back(re);
        node[top++] = nod;
    }
}

char haha(int x)
{
    char c = node[x].ve[0];
    int len = node[x].ve.size();
    for(int i=0; i<len; i++)
    {
        if(node[x].ve[i] == 'A' || node[x].ve[i] == 'B') c = node[x].ve[i];
    }
    return c;
}

void dfs(int x)
{
    if(node[x].left!=-1)
    {
        book[x] = 1;
        dfs(node[x].left);
        dfs(node[x].right);
    }
}

int main()
{
    int n, i, ll, rr;
    char s[15];
    cin >> n;
    top = 0;
    memset(book, 0, sizeof(book));
    for(i=0;i<n;i++)
    {
        cin >> s;
        ll = add_node(s[2]);
        rr = add_node(s[4]);
        add_op(s[0], s[3], ll, rr);
    }
    for(i=0;i<top;i++)
    {
        if(node[i].left==-1) continue;
        re[i][0] = haha(i);
        re[i][1] = '=';
        Node le, ri;
        le = node[node[i].left];
        ri = node[node[i].right];
        re[i][2] = le.left==-1 ? le.id : haha(node[i].left);
        re[i][3] = node[i].id;
        re[i][4] = ri.right==-1 ? ri.id : haha(node[i].right);
        re[i][5] = '';
    }
    for(i=top-1;i>=0;i--)
    {
        if(re[i][0]=='A')
        {
            dfs(i);
            break;
        }
    }
    for(i=top-1;i>=0;i--)
    {
        if(re[i][0]=='B')
        {
            dfs(i);
            break;
        }
    }
    for(i=0;i<top;i++)
    {
        if(book[i]) cout << re[i] << endl;
    }
    return 0;
}