基于cs50库实现类Python输入函数：如何支持多数据类型？

首先，我理解你想在C里做一个类似Python input()的通用函数，既能输出提示文本，又能根据需要获取不同类型的输入——就像CS50自带的get_int()、get_float()那样，但把提示和输入整合到一个函数里。

因为C是静态类型语言，不像Python那样支持动态类型推导，所以我们没法直接写一个万能的input()函数自动适配所有类型，但可以通过两种常用方式来模拟这种多类型支持：

方案1：用宏生成特定类型的输入函数（推荐，类型安全） #

这种方式和CS50的get_*函数风格一致，通过宏自动生成对应类型的输入函数，每个函数负责处理一种类型，调用起来直观且类型安全。

代码实现 #

#include <stdio.h>
#include <cs50.h>

// 宏定义：自动生成指定类型的输入函数
// 参数说明：
// - type: 目标数据类型（如int、float）
// - func_name: 生成的函数名（如input_int）
// - cs50_get_func: 对应的CS50输入函数（如get_int）
#define DEFINE_INPUT_FUNC(type, func_name, cs50_get_func) \
type func_name(const char *prompt) { \
    printf("%s", prompt); \
    return cs50_get_func(""); \
}

// 生成int和float类型的输入函数
DEFINE_INPUT_FUNC(int, input_int, get_int)
DEFINE_INPUT_FUNC(float, input_float, get_float)

// 单独处理字符串类型（避免宏处理指针的语法问题）
char* input_string(const char *prompt) {
    printf("%s", prompt);
    return get_string("");
}

int main() {
    // 测试字符串输入
    char *name = input_string("What's your name? ");
    printf("Hello, %s!\n", name);

    // 测试整数输入
    int age = input_int("How old are you? ");
    printf("You're %d years old.\n", age);

    // 测试浮点数输入
    float weight = input_float("What's your weight (kg)? ");
    printf("Your weight is %.1f kg.\n", weight);

    // 注意：CS50的get_string返回堆内存，需要手动释放
    free(name);
    return 0;
}

优点 #

• 完全类型安全，编译时就能检查类型错误
• 调用方式简洁，和CS50原生函数的使用逻辑一致
• 不需要额外的内存管理（除了字符串需要free，和CS50的规则一致）

方案2：用枚举+函数指针实现通用输入函数（更接近Python风格） #

如果你想要一个更像Python input()的单一函数，通过参数指定输入类型，可以用枚举标记类型，结合void*返回值实现。但要注意这种方式需要手动处理类型转换和内存释放。

代码实现 #

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

// 定义支持的输入类型枚举
typedef enum {
    INPUT_INT,
    INPUT_FLOAT,
    INPUT_STRING
} InputType;

// 通用输入函数：根据指定类型返回对应值的指针
void* input(const char *prompt, InputType type) {
    printf("%s", prompt);
    switch(type) {
        case INPUT_INT: {
            // 为整数分配堆内存
            int *int_val = malloc(sizeof(int));
            *int_val = get_int("");
            return int_val;
        }
        case INPUT_FLOAT: {
            // 为浮点数分配堆内存
            float *float_val = malloc(sizeof(float));
            *float_val = get_float("");
            return float_val;
        }
        case INPUT_STRING:
            // CS50的get_string已经返回堆内存的字符串
            return get_string("");
        default:
            return NULL;
    }
}

int main() {
    // 获取字符串输入
    char *name = input("What's your name? ", INPUT_STRING);
    printf("Hello, %s!\n", name);

    // 获取整数输入
    int *age = input("How old are you? ", INPUT_INT);
    printf("You're %d years old.\n", *age);

    // 获取浮点数输入
    float *height = input("What's your height (m)? ", INPUT_FLOAT);
    printf("Your height is %.2f m.\n", *height);

    // 手动释放所有堆内存
    free(name);
    free(age);
    free(height);
    return 0;
}

注意事项 #

• 需要手动将void*转换为对应类型的指针，编译时不会检查类型错误，容易出错
• 除了字符串（由CS50的get_string分配），整数和浮点数的内存需要手动free，避免内存泄漏

总的来说，方案1更符合C语言的静态类型特性，也更适配CS50的开发环境，推荐优先使用。

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