SDUT 简单的代码生成程序

SDUT 简单的代码生成程序

Description

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

Input

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

Output

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

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;
string s[200];
char jcq[200];
int n, m, cnt;

int use(int i, char c)
{
    for (; i < n; i++)
        if (s[i][3] == c || s[i][5] == c)
            return i;
    return n;
}

int find(int pos)
{
    if (cnt < m)
        return cnt++;
    int Max = -1, id;
    for (int i = 0; i < cnt; i++)
    {
        if (use(pos, jcq[i]) < Max)
            continue;
        Max = use(pos, jcq[i]);
        id = i;
    }
    return id;
}

signed main()
{
    map<char, string> mp;
    mp['+'] = "ADD ", mp['-'] = "SUB ", mp['*'] = "MUL ", mp['/'] = "DIV ";

    cin >> n >> m;
    for (int i = 0; i < n; i++)
        cin >> s[i];
    for (int i = 0; i < n; i++)
    {
        int id1 = find(jcq, jcq + cnt, s[i][3]) - jcq;
        if (id1 == cnt)
        {
            id1 = find(i);
            if (jcq[id1] != '\0' && use(i, jcq[id1]) < n)
            {
                printf("ST R%d, %c\n", id1, jcq[id1]);
                jcq[id1] = '\0';
            }
            printf("LD R%d, %c\n", id1, s[i][3]);
        }
        cout << mp[s[i][4]];
        int id2 = find(jcq, jcq + cnt, s[i][5]) - jcq;
        if (id2 == cnt)
            printf("R%d, %c\n", id1, s[i][5]);
        else
            printf("R%d, R%d\n", id1, id2);
        jcq[id1] = s[i][0];
    }
}