L3-023 计算图

建立结构体保存每个结点的前驱，操作符，来回两遍拓扑排序~

#include<bits/stdc++.h>
using namespace std;
const int maxn=;
struct node {
    vector<int> pre;
    double data;
    int fuhao=;
}Node[maxn];
vector<int> g[maxn];
int a[maxn];
int inDegree[maxn];
int N;
stack<int> topOrder;
double getnum (node a) {
    if (a.pre.size()==) return a.data;
    else if (a.pre.size()==) {
        if (a.fuhao==) return exp(Node[a.pre[]].data);
        else if (a.fuhao==) return log(Node[a.pre[]].data);
        else if (a.fuhao==) return sin(Node[a.pre[]].data);
    }
    else if (a.pre.size()==) {
        if (a.fuhao==) return Node[a.pre[]].data+Node[a.pre[]].data;
        else if (a.fuhao==) return Node[a.pre[]].data-Node[a.pre[]].data;
        else if (a.fuhao==) return Node[a.pre[]].data*Node[a.pre[]].data;
    }
}
double topSort () {
    queue<int> q;
    for (int i=;i<N;i++)
    if (inDegree[i]==) q.push(i);
    while (!q.empty()) {
        int u=q.front();
        q.pop();
        topOrder.push(u);
        Node[u].data=getnum(Node[u]);
        for (int i=;i<g[u].size();i++)
        if (--inDegree[g[u][i]]==) q.push(g[u][i]);
        if (q.empty()) return Node[u].data;
    }
}
double dfs (int u,int x) {
    if (Node[u].fuhao==) {
        if (u==x) return ;
        else return ;
    }
    else if (Node[u].fuhao==) return dfs(Node[u].pre[],x)+dfs(Node[u].pre[],x);
    else if (Node[u].fuhao==) return dfs(Node[u].pre[],x)-dfs(Node[u].pre[],x);
    else if (Node[u].fuhao==) return dfs(Node[u].pre[],x)*Node[Node[u].pre[]].data+dfs(Node[u].pre[],x)*Node[Node[u].pre[]].data;
    else if (Node[u].fuhao==) return exp(Node[Node[u].pre[]].data)*dfs(Node[u].pre[],x);
    else if (Node[u].fuhao==) return 1.0/Node[Node[u].pre[]].data*dfs(Node[u].pre[],x);
    else if (Node[u].fuhao==) return cos(Node[Node[u].pre[]].data)*dfs(Node[u].pre[],x);
}
int main () {
    scanf ("%d",&N);
    int fuhao;
    double x,y;
    vector<int> v1;
    for (int i=;i<N;i++) {
        scanf ("%d",&fuhao);
        if (fuhao==) {
            scanf ("%lf",&x);
            Node[i].data=x;
            v1.push_back(i);
        }
        else if (fuhao>=&&fuhao<=) {
            scanf ("%lf %lf",&x,&y);
            Node[i].pre.push_back((int)x);
            Node[i].pre.push_back((int)y);
            inDegree[i]+=;
            g[(int)x].push_back(i);
            g[(int)y].push_back(i);
            Node[i].fuhao=fuhao;
        }
        else if (fuhao>=&&fuhao<=) {
            scanf ("%lf",&x);
            Node[i].pre.push_back((int)x);
            inDegree[i]+=;
            g[(int)x].push_back(i);
            Node[i].fuhao=fuhao;
        }
    }
    printf ("%.3f\n",topSort());
    int u=topOrder.top();
    for (int i=;i<v1.size();i++) {
        if (i!=) printf (" ");
        printf ("%.3f",dfs(u,v1[i]));
    }
    return ;
}