3.5.3 for 循环 的算法有问题

[feiquan123]

func LoopAdd(cnt, v0, step int) int {
    result := v0
    for i := 0; i < cnt; i++ {
        result += step
    }
    return result
}

比如 1+2+...+100 等差数列可以这样计算 LoopAdd(100, 1, 1)，而 10+8+...+0 等差数列则可以这样计算 LoopAdd(5, 10, -2)。

问题描述

按这个算法 LoopAdd(100, 1, 1) = 101 期望：5050 LoopAdd(5, 10, -2) = 0 期望：30

[feiquan123]

修改后：

func LoopAdd(cnt, v0, step int) int {
    result := v0
    next := v0
    for i := 1; i < cnt; i++ {
        next += step
        result += next
    }
    return result
}

汇编：

TEXT ·LoopAdd(SB),$0-32
    MOVQ cnt+8*0(FP), AX // cnt
    MOVQ v0+8*1(FP), BX // v0 /result
    MOVQ step+8*2(FP), CX // step

    MOVQ BX, R8 // next
    MOVQ $1, DX // i

LOOP_IF:
    CMPQ DX, AX // compare i, cnt
    JL LOOP_BODY // if i < cnt; goto LOOP_BODY
    JMP LOOP_END

LOOP_BODY:
    ADDQ CX, R8 // next += step
    ADDQ R8, BX // result += next
    ADDQ $1, DX // i++
    JMP LOOP_IF

LOOP_END:
    MOVQ BX, ret+8*3(FP) // return result
    RET

[chai2010]

发现 BUG 直接上 PR