DAG优化

DAG优化

SDUT OJ DAG优化
DAG优化
Time Limit: 1000 ms Memory Limit: 65536 KiB

Problem 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
Sample Output
B=B+B
A=C+C

#include <bits/stdc++.h>
using namespace std;
int n;
int cnt = 0;
char s[10];
char ans[110][10]; //存输出内容
bool flag[110];
//构建一个结构体来存储DAG图
struct Node
{
    //存储变量
    char id;
    //存储左右儿子
    int left = -1, right = -1;
    vector<char> var;//标记变量
} node[110];

bool ifind_var(int i, char c)
{
    for(int j = 0; j < node[i].var.size(); j++)
    {
        if(node[i].var[j] == c)
            return true;
    }
    return false;
}
int add_node(char c)
{
    //找一下图中是否已经存在该节点
    //若存在，直接返回位置
    for(int i = cnt-1; i >= 0; --i)
        if(node[i].id == c && ifind(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++;
}

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

int main()
{
    scanf("%d",&n);
    //构建DAG图
    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);
    }
    //把DAG图中的内容放到一个输出数组中
    for(int i = 0; i < cnt; ++i)
    {
        if(node[i].left != -1)
        {
            //第一个元素即为该节点的标记变量
            ans[i][0] = node[i].var[0];
            //第二个都是'='
            ans[i][1] = '=';
            Node ll = node[node[i].left];
            Node 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;
            //标记结束
            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;
}