P - 简单的代码生成程序

P - 简单的代码生成程序

Description
通过三地址代码序列生成计算机的目标代码,在生成算法中,对寄存器的使用顺序为:寄存器中存有 > 空寄存器 > 内存中存有 > 以后不再使用 > 最远距离使用

Input
单组输入,给定输出的三地址代码的个数和寄存器的个数.所有的变量为大写字母,寄存器的数量不超过9

Output
参照示例格式输出,不需要将最后的寄存器中的值写回内存

不再使用变量不用写回内存

Sample
Input
4 2
T:=A-B
U:=A-C
V:=T+U
W:=V+U
Output
LD R0, A
SUB R0, B
LD R1, A
SUB R1, C
ADD R0, R1
ADD R0, R1

#include <bits/stdc++.h>
using namespace std;
char s[110][1100],p[10]={0};
int n,m,top=0;
int get(char ch){//是否在寄存器中
    for(int i=0;i<m;i++)
        if(ch==p[i])
            return i;
    return -1;
}
int use(int x,char ch){//是否在之后的语句中会用到
    for(int i=x;i<n;i++)
        if(ch==s[i][3]||ch==s[i][5])
            return i;
    return n;
}
int find(int x){//使用哪个寄存器存值
    if(top<m)
        return top++;
    int t=-1,max=-1;
    for(int i=0;i<m;i++){
        int k=use(x,p[i]);
        if(k>max){
            max=k;
            t=i;
        }
    }
    return t;
}
void print1(char ch){
    if(ch=='+')cout<<"ADD ";
    else if(ch=='-')cout<<"SUB ";
    else if(ch=='*')cout<<"MUL ";
    else if(ch=='\\')cout<<"DIV ";
}
void print2(char ch){
    int x=get(ch);
    if(x!=-1)
        cout<<"R"<<x<<endl;
    else
        cout<<ch<<endl;
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
        cin>>s[i];
    for(int i=0;i<n;i++){
        int x=get(s[i][3]);
        if(x==-1){
            x=find(i);
            if(p[x]!='\0'&&use(i,p[x])<n){
                cout<<"ST R"<<x<<", "<<p[x]<<endl;
                p[x]=NULL;
            }
            cout<<"LD R"<<x<<", "<<s[i][3]<<endl;
        }
        print1(s[i][4]);
        cout<<"R"<<x<<", ";
        print2(s[i][5]);
        p[x]=s[i][0];
    }
    return 0;
}