基于二叉树的算术表达式计算与实现

非常有趣的一道题:
代码参考 https://blog.csdn.net/over786/article/details/17395567
把一个正常的表达式写成这个样子咯
问题的难点在于整数有多位,而且你要判断好哪个符号作为根
比如5+3*4
基于二叉树的算术表达式计算与实现_第1张图片

基于二叉树的算术表达式计算与实现_第2张图片

找根,你就按照正常方法算
5+4*3-2
那这个减号最后计算,作为根
其他的递归跑一次就行啦
括号里的+-*/都不应该成为最后计算的
最右边的加号作为根
我的除法用的是整数除法,你要是不喜欢可以在check函数里用double类型
the end
参考表达式示例
(3+4)*(4+412)
1+2*3-5/4

小bug 解决 处理连续括号如((4+3))

#include 
#include 
#include 
#include 
#define MAX 0x3f3f3f3f
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
typedef struct  TNode{
    int flag;
    int data;//flag=0 
    char ch;//flag=1
    struct TNode * lChild;//左孩子 
    struct  TNode * rChild;//右孩子 
};
//(5+3*4)*2/3-2*5
int cal(struct TNode *root){
    if(root->flag==1){
        switch(root->ch){
            case '+':{
                return cal(root->lChild)+cal(root->rChild);
                break;
            }
            case '-':{
                return cal(root->lChild)-cal(root->rChild);
                break;
            }
            case '/':{
                return cal(root->lChild)/cal(root->rChild);
                break;
            }
            case '*':{
                return cal(root->lChild)*cal(root->rChild);
                break;
            }
        }
    }
    return root->data;
}
int check(char s[],int start,int end){
    int i;
    int sum=0;
    int flag=1;
    if(s[start]=='-'){
        flag=-1;
        start++;
    }
    for(i=start;i<=end;i++){
        if(!isdigit(s[i])) return MAX;
        sum=sum*10+s[i]-'0';
    }
    return sum*flag;
}
void postOrder(struct TNode *root){
    if(root){
        postOrder(root->lChild);
        postOrder(root->rChild);
        if(root->flag==0){
            printf("%d ",root->data);
        }
        else{
            printf("%c ",root->ch);
        }       
    }
} 
struct TNode * buildTree(char s[],int start,int end){
    struct TNode * root=(struct TNode *)malloc(sizeof(struct TNode));
    int cnt=0;
    int m;
    int i;
    if(start>end) return NULL;
    int posPlusOrSub=0;//加减号位置 
    int numPlusOrSub=0;//加减号个数 
    int posDivOrMul=0;//乘除号位置 
    int numDivOrMul=0;//乘除号个数 
    int num;
    num=check(s,start,end);     
    if(num!=0x3f3f3f3f){//只有数字 
        root->flag=0;
        root->data=num;
        root->lChild=NULL;
        root->rChild=NULL;
        return root;
    }
    //有操作符 
    int in_brackets=0;//不在括号里的标识符 
    for(int k=start;k<=end;k++){
        if(s[k]=='('){
            in_brackets++;
        }
        else if(s[k]==')'){
            in_brackets--;
        }
        if(!in_brackets){//括号之外 
            if(s[k]=='+'||s[k]=='-'){
                posPlusOrSub=k;
                numPlusOrSub++;
            }
            else if(s[k]=='*'||s[k]=='/'){
                posDivOrMul=k;//乘除号位置 
                numDivOrMul++;//乘除号个数 
            }
        }
    }
    int pos_root;
    //寻找根节点 有加减用加减没加减用乘除 
    if(numPlusOrSub){
        pos_root=posPlusOrSub;
    }
    else if(numDivOrMul){
        pos_root=posDivOrMul;
    }
    else{//找不到根 递归再找一次 
        return buildTree(s,start+1,end-1);
    }
    root->flag=1;
    root->ch=s[pos_root];
    root->lChild = buildTree(s,start,pos_root-1);
    root->rChild = buildTree(s,pos_root+1,end);
    return root;
}
int main(int argc, char** argv) {
    while(1){
        char a[200];
        printf("请输入表达式:");
        scanf("%s",a);
        printf("你输入的表达式为:%s\n",a);
        struct TNode* b=(struct TNode *)malloc(sizeof(struct TNode));
        b=buildTree(a,0,strlen(a)-1);
        printf("Your result is %d\n",cal(b));   
    }
    return 0;
}

你可能感兴趣的:(数据结构)