编写MIPS汇编函数实现数组交替累加连续升/降数值

需求说明 #

给定数组A及整数a、b、n，实现以下逻辑：

• 若a < b：数组前2n个元素依次交替累加a、b（即A[0]+=a，A[1]+=b，A[2]+=a...共处理2n个元素）
• 若a >= b：数组前2n个元素依次交替累加b、a（即A[0]+=b，A[1]+=a，A[2]+=b...共处理2n个元素）

已定义寄存器分配：

• $a0 = a
• $a1 = b
• $a2 = n
• $a3 = 数组A的基地址

完整汇编代码 #

# 计算循环终止地址：数组A的第2n个元素之后的地址（32位元素占4字节，2n个元素总字节数8n）
mul $t0, $a2, 8
add $t0, $t0, $a3
xor $t1, $t1, $t1       # 初始化当前元素的字节偏移量为0（对应A[0]）
slt $t2, $a0, $a1       # 判断a < b：成立则$t2=1，否则$t2=0
bne $t2, $0, foo1       # a < b时跳转到foo1分支执行

# ---------------------- foo2分支：a >= b，交替累加b、a ----------------------
foo2:
    add $t3, $a3, $t1   # 计算当前元素的内存地址
    lw $t4, 0($t3)      # 读取当前元素值
    add $t4, $t4, $a1   # 累加b到当前元素
    sw $t4, 0($t3)      # 将结果写回数组
    
    addi $t1, $t1, 4    # 偏移量+4，指向下一个元素
    beq $t1, $t0, exit  # 若已处理完所有元素，跳转到退出逻辑
    
    add $t3, $a3, $t1   # 计算下一个元素的内存地址
    lw $t4, 0($t3)      # 读取元素值
    add $t4, $t4, $a0   # 累加a到该元素
    sw $t4, 0($t3)      # 写回数组
    
    addi $t1, $t1, 4    # 偏移量+4
    bne $t1, $t0, foo2  # 未处理完则继续循环
    j exit

# ---------------------- foo1分支：a < b，交替累加a、b ----------------------
foo1:
    add $t3, $a3, $t1   # 计算当前元素的内存地址
    lw $t4, 0($t3)      # 读取当前元素值
    add $t4, $t4, $a0   # 累加a到当前元素
    sw $t4, 0($t3)      # 写回数组
    
    addi $t1, $t1, 4    # 偏移量+4，指向下一个元素
    beq $t1, $t0, exit  # 已处理完则退出
    
    add $t3, $a3, $t1   # 计算下一个元素的内存地址
    lw $t4, 0($t3)      # 读取元素值
    add $t4, $t4, $a1   # 累加b到该元素
    sw $t4, 0($t3)      # 写回数组
    
    addi $t1, $t1, 4    # 偏移量+4
    bne $t1, $t0, foo1  # 未处理完则继续循环

# 函数退出逻辑
exit:
    jr $ra

关键逻辑说明 #

1. 终止地址计算：32位MIPS中每个整数占4字节，2n个元素总字节数为8n，用mul指令计算该值后，加上数组基地址得到循环终止的内存地址，避免数组越界。
2. 分支判断：用slt指令判断a和b的大小，分别进入对应分支执行交替累加逻辑。
3. 循环处理：每个分支每次处理两个元素（完成一次a/b或b/a的交替），更新偏移量后判断是否已处理完2n个元素，未完成则继续循环。
4. 寄存器使用：用临时寄存器$t0-$t4存储中间值，不破坏输入参数寄存器$a0-$a3。

内容的提问来源于stack exchange，提问作者Prime_Number