请问这个程序在什么时候不能正确得出结果，应该怎么改

Question

提交程序后有一个结果不对，但不知道哪种情况，谁能帮忙解决一下

Answer 1

我觉得有一种可能是如下数据： 2 2 1 0 答案 ： 7，这样子的话会导致数据越界的，其实不用数组也是行滴

#include<stdio.h>

int main()
{
    int ans = 0; int sum = 0; int n;
    
    while(true)
    {
        scanf("%d",&n);

        if(n == 0) break;
        
        else if(n == 1) ans ++ , sum = 0;
        
        else ans = ans + sum + 2 , sum += 2; 
    }
    
    printf("%d",ans);
}