C语言命令行输入转整数及双输入模式程序报错排查

你的第一段代码存在几个关键问题：

• 参数访问错误：atoi(argv)完全错误，argv是指向字符串数组的指针（char**），而atoi需要传入单个字符串（char*）。而且如果没有命令行参数，argv[1]是NULL，直接访问会触发未定义行为。
• 逻辑判断错误：用atoi(argv)==0来判断是否有命令行输入不成立——如果用户确实输入了0作为命令行参数，这个条件也会触发运行时输入的分支，无法区分“无参数”和“参数为0”。
• 未检查参数数量：应该先通过argc判断是否有命令行参数（argc > 1表示有额外参数）。

修正后的代码：

#include <stdio.h>
#include <stdlib.h>

int main(int argc, char **argv)    
{   
    int num;
    // 先判断是否有命令行参数
    if (argc > 1) {
        // 将第一个命令行参数转为整数
        num = atoi(argv[1]);
    } else {
        printf("Enter the number : ");
        scanf("%d", &num);
    }
    // 验证输入结果
    printf("You entered: %d\n", num);
    return 0;
}

第二段代码的核心问题集中在参数判断、循环逻辑、内存管理和计算逻辑上：

• 参数访问风险：当没有命令行参数时，argv[1]是NULL，调用atoi(argv[1])会导致未定义行为。
• 循环范围错误：命令行参数的循环for (i = 2; i<=argc; i++)会越界，argc是参数总数，argv[argc]是NULL，正确的循环应该遍历从argv[1]到argv[argc-1]的所有参数。
• 整数除法问题：avg = sum / num_of_values是整数除法，结果会截断小数部分，需要强制转换为浮点型计算。
• 内存泄漏：运行时输入分支中malloc的sum_arr没有释放，会造成内存泄漏。
• 边界情况未处理：如果用户输入的num_of_values为0，或者没有任何命令行参数且运行时输入0个值，会触发除以0的错误。

修正后的代码：

#include <stdio.h>
#include <stdlib.h>

int main(int argc, char **argv)    
{   
    int num_of_values = 0, sum = 0;
    int i;
    float avg = 0.0f;
    int* sum_arr = NULL;

    // 判断是否有命令行参数（argv[0]是程序名，argc>1表示有输入值）
    if (argc > 1) {
        num_of_values = argc - 1;
        for (i = 1; i < argc; i++) {
            sum += atoi(argv[i]);
        }
    } else {
        printf("Enter the number of values: ");
        // 检查输入是否有效
        if (scanf("%d", &num_of_values) != 1 || num_of_values <= 0) {
            printf("Invalid input for number of values.\n");
            return 1;
        }
        sum_arr = malloc(num_of_values * sizeof(int));
        // 检查内存分配是否成功
        if (sum_arr == NULL) {
            printf("Memory allocation failed.\n");
            return 1;
        }
        printf("Enter the values: ");
        for(i=0; i<num_of_values; i++) {
            // 检查每个值的输入是否有效
            if (scanf("%d", &sum_arr[i]) != 1) {
                printf("Invalid value input.\n");
                free(sum_arr);
                return 1;
            }
            sum += sum_arr[i];
        }
        // 释放分配的内存
        free(sum_arr);
    }

    // 避免除以0
    if (num_of_values == 0) {
        printf("No values provided.\n");
        return 1;
    }

    // 强制转换为浮点型计算平均值
    avg = (float)sum / num_of_values;
    printf("Sum and Average is %d and %.2f\n", sum, avg);
    
    return 0;
}

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