吴楚月的表达式(dfs+维护运算同优先级块）

最新推荐文章于 2021-06-11 20:27:22 发布

小菜鸡加油

最新推荐文章于 2021-06-11 20:27:22 发布

阅读量262

点赞数

分类专栏：思维 # 树上dfs

本文链接：https://blog.csdn.net/zstuyyyyccccbbbb/article/details/113872445

版权

思维同时被 2 个专栏收录

514 篇文章 4 订阅

订阅专栏

树上dfs

30 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用深度优先搜索（DFS）解决二叉树结构的数学表达式计算问题。作者探讨了如何维护临时变量`tmp[]`来处理乘除运算级联通块的值，并在遇到加减运算时正确地更新结果。文章还提到，虽然初看可能认为解决方案的时间复杂度为O(n)，但实际上由于沿着树链跳跃，复杂度为O(logn)。

摘要由CSDN通过智能技术生成

https://ac.nowcoder.com/acm/contest/9984/H

思路:

虽然上了数据结构课的第一眼可能是要用个栈...（栈也确实能做

刚开始想的时候就dp[i]维护到i这个节点的答案，然后dfs跑一遍，发现有前一个符号的加减关系，于是开一个lastop，dfs传一个父亲的字符。

但是还不行。因为比如1+2*3/4*5。dp[4]这个答案是不能用于转移的，答案要分割成+号前面的一个块和5前面的联通块。

比赛的时候想错了，以为暴力往上跳是O（n)级别的。但是实际上是在一根树链往上跳的，所以类似的思想，是logn的。

于是tmp[i]维护到i的*/联通块的值。

注意tmp[i]的起点是其op="+“"-"的。所以其开始的维护是在判str[i]=='+'’-‘那里维护的。

~~二叉树计算器的新解法(bushi，之前是对建树符号重新排序做的~~

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e5+100;
typedef long long LL;
const LL mod=1e9+7;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
LL val[maxn],fa[maxn];
char str[maxn];
LL ans[maxn],tmp[maxn];///tmp[]:维护*/运算级联通块的值
LL ksm(LL a,LL k){LL res=1;while(k>0){if(k&1) res=res*a%mod;k>>=1;a=a*a%mod;}return res%mod;}
void dfs(LL u,LL f){
    if(u==1) ans[u]=tmp[u]=val[u];
    else{
        if(str[u]=='+'){
            ans[u]=(ans[fa[u]]%mod+val[u]%mod)%mod;     tmp[u]=val[u];
        }
        else if(str[u]=='-'){
            ans[u]=(ans[fa[u]]%mod-val[u]%mod+mod)%mod; tmp[u]=val[u];
        }
        else if(str[u]=='*'){
            LL t=u;
            tmp[u]=(tmp[fa[u]]%mod*val[u]%mod)%mod;
            while(str[t]=='*'||str[t]=='/') t=fa[t];
            if(str[t]=='+'){
                ans[u]=(ans[fa[t]]%mod+tmp[u]%mod)%mod;
            }
            else if(str[t]=='-'){
                ans[u]=(ans[fa[t]]%mod-tmp[u]%mod+mod)%mod;
            }
            else ans[u]=tmp[u];
        }
        else if(str[u]=='/'){
            LL t=u;
            tmp[u]=(tmp[fa[u]]%mod*ksm(val[u],mod-2)%mod)%mod;
            while(str[t]=='*'||str[t]=='/') t=fa[t];
            if(str[t]=='+'){
                ans[u]=(ans[fa[t]]%mod+tmp[u]%mod)%mod;
            }
            else if(str[t]=='-'){
                ans[u]=(ans[fa[t]]%mod-tmp[u]%mod+mod)%mod;
            }
            else ans[u]=tmp[u];
        }
    }
    for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(v==f) continue;
        dfs(v,u);
    }
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL n;cin>>n;
  for(LL i=1;i<=n;i++) cin>>val[i];
  for(LL i=2;i<=n;i++) cin>>fa[i];
  for(LL i=2;i<=n;i++) cin>>str[i];
  for(LL i=2;i<=n;i++){
    LL u=fa[i];
    g[u].push_back(i);
  }
  dfs(1,0);
  for(LL i=1;i<=n;i++){
    cout<<ans[i]<<" ";
  }
  cout<<"\n";
return 0;
}