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 r[100],a[100][100];
int n, m, top = 0;
int get(char ch)
{
    int i;
    for(i = 0; i < m; i++)
    {
        if(ch == r[i])
        {
            return i;
        }
    }
    return -1;
}
int use(int x, char ch)
{
    int i;
    for(i = x; i < n; i++)
    {
        if(ch == a[i][3] || ch == a[i][5])
        {
            return i;
        }
    }
    return n;
}
int main()
{
    int i,j;
    cin>>n>>m;
    for(i = 0; i < n; i++)
    {
        cin>>a[i];
    }
    for(i = 0; i < n; i++)
    {
        int id = get(a[i][3]);
        if(id == -1)
        {
            if(top < m)
            {
                id=top++;
            }
            else
            {
                int k,ans = -1;
                for(j = 0; j < m; j++)
                {
                    k = use(i, r[j]);
                    if(k > ans)
                    {
                        ans = k;
                        id = j;
                    }
                }
                if(ans < n)
                {
                    cout<<"ST R"<<id<<", "<<r[id]<<endl;
                    r[id] = NULL;//???
                }
            }
            cout<<"LD R"<<id<<", "<<a[i][3]<<endl;
        }
        char ch=a[i][4];
        if(ch == '+')
        {
            cout<<"ADD ";
        }
        else if(ch == '-')
        {
            cout<<"SUB ";
        }
        else if(ch == '*')
        {
            cout<<"MUL ";
        }
        else if(ch == '\\')
        {
            cout<<"DIV ";
        }
        cout<<"R"<<id<<", ";
        int id1 = get(a[i][5]);
        if(id1 == -1)
        {
            cout<<a[i][5]<<endl;
        }
        else
        {
            cout<<"R"<<id1<<endl;
        }
        r[id] = a[i][0];
    }
    return 0;
}