7-5 简单的代码生成程序-A (10 分)

7-5 简单的代码生成程序-A (10 分)
通过三地址代码序列生成计算机的目标代码,在生成算法中,对寄存器的使用顺序为:寄存器中存有 > 空寄存器 > 内存中存有 > 以后不再使用 > 最远距离使用

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

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

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

输入样例:
在这里给出一组输入。例如：

4 2
T:=A-B
U:=A-C
V:=T+U
W:=V+U

输出样例:
在这里给出相应的输出。例如：

LD R0, A
SUB R0, B
LD R1, A
SUB R1, C
ADD R0, R1
ADD R0, R1

作者: sdutacm
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

/*******************
此程序将三元式代码翻译为汇编语言代码。
*******************/
#include <bits/stdc++.h>
using namespace std;
char s[100][100];
char reg[101];
int n,m,mm = 0;
int is_register(char ch)
{
    for(int i = 0; i < m; i++)
    {
        if(ch == reg[i])
        {
            return i;
        }
    }
    return -1;
}
int last_use(int r,char ch)
{
    for(int  i = r; i < n; i++)
    {
        if(ch == s[i][3]||ch == s[i][5])
            return i;
    }
    return n;
}
int get_register(int r)
{
    if(mm < m)return mm++;
    int f = -1,ans = -1;
    for(int i = 0; i < m; i++)
    {

        int v = last_use(r,reg[i]);
        if(v > ans)
        {
            ans = v;
            f = i;
        }
    }
    return f;
}
void print1(char ch)
{
    if(ch == '+')printf("ADD ");
    else if(ch == '-')printf("SUB ");
    else if(ch == '*')printf("MUL ");
    else if(ch == '\\')printf("DIV ");
}
void print2(char ch)
{
    int r = is_register(ch);
    if(r != -1)
        printf("R%d\n",r);
    else printf("%c\n",ch);
}
int main()
{
    cin>>n>>m;
    for(int i = 0; i < n; i++)
    {
        cin>>s[i];
    }
    for(int i = 0; i < n; i++)
    {
        int r = is_register(s[i][3]);
        if(r == -1)
        {
             r = get_register(i);
             if(reg[r] != '\0'&&last_use(i,reg[r]) < n)
             {
                 printf("ST R%d, %c\n",r,reg[r]); ///把寄存器r里面的东东弄到内存里去
                 reg[r] = NULL;
             }
             printf("LD R%d, %c\n",r,s[i][3]);
        }
        print1(s[i][4]);
        printf("R%d, ",r);
        print2(s[i][5]);
        reg[r] = s[i][0];
    }
    return 0;
}