memset函数

1. 基本介绍

   • memset函数是C/C++语言中的一个标准库函数，用于将一段内存空间的每个字节设置为指定的值。它定义在<string.h>（C语言）或<cstring>（C++语言）头文件中。

2. 函数原型及参数含义

   • 函数原型为void *memset(void *s, int c, size_t n);。
   • 其中，s是指向要填充的内存块的起始地址的指针。这个内存块可以是数组、动态分配的内存或者其他连续的内存区域。例如，对于一个字符数组char arr[10];，可以使用arr作为s的参数。
   • c是要设置的字节值，这个值会被截断为无符号字符型（unsigned char），所以其取值范围实际上是0 - 255。例如，如果想将内存区域设置为0，就传入0；如果想设置为'\xff'（十六进制的FF），就传入0xff（在C/C++中，十六进制数以0x开头）。
   • n是要填充的字节数。例如，如果有一个int数组int arr[5];，每个int通常占4个字节（假设32位系统），如果要将整个数组初始化为0，就需要填充5 * sizeof(int)个字节。

3. 简单示例

   • 初始化字符数组
     • 以下代码使用memset函数将一个字符数组初始化为0：

     #include <iostream>
     #include <cstring>
     int main()
     {
         char arr[10];
         memset(arr, 0, sizeof(arr));
         for (int i = 0; i < 10; ++i)
         {
             std::cout << (int)arr[i] << " ";
         }
         std::cout << std::endl;
         return 0;
     }
     

     • 在这个例子中，memset函数将arr数组的10个字节全部设置为0。然后通过循环输出数组每个元素的值（将字符强制转换为整数类型以查看其ASCII码值），可以看到输出的都是0。
   • 初始化整数数组部分元素
     • 假设我们有一个整数数组，想将其前一部分元素初始化为某个值。以下代码将一个int数组的前6个字节（相当于一个半int元素，假设int占4字节）初始化为-1：

     #include <iostream>
     #include <cstring>
     int main()
     {
         int arr[5];
         memset(arr, 0xff, 6);
         for (int i = 0; i < 5; ++i)
         {
             std::cout << arr[i] << " ";
         }
         std::cout << std::endl;
         return 0;
     }
     

     • 在这里，由于int通常占4字节，我们只填充了6字节，所以第一个int元素的前两个字节被设置为0xff，后两个字节为0（因为没有填充到），在小端序（数据的低位字节存于低地址）系统下，这个int元素会被解释为一个负数（具体的值取决于字节序和编译器实现），后面的int元素则是未定义的行为，因为没有完全填充。

4. 注意事项

   • 数据类型和字节数：
     • 在使用memset函数时，需要特别注意数据类型的字节数。例如，对于非char类型的数组（如int、double等），如果想要正确地初始化整个数组，必须填充正确的字节数，即n的值应该是数组元素个数 * sizeof(数组元素类型)。如果填充的字节数不足或过多，可能会导致意想不到的结果。
   • 适用于字节操作：
     • memset函数是按照字节进行操作的。所以对于复杂的数据结构或者非字节对齐的数据类型，使用memset可能会有问题。例如，对于包含指针成员的结构体，使用memset将其全部初始化为0可能会导致指针被错误地设置为NULL，除非这是期望的行为。在这种情况下，更安全的做法可能是使用构造函数（在C++中）或者手动初始化每个成员。

5. 算法比赛中的初始化值

   • 初始化数组为0：
     • 在许多算法问题中，经常需要将数组初始化为0。例如，在计数问题中，可能会使用一个数组来记录某个元素出现的次数。像动态规划问题，若定义一个状态数组dp，通常会将其初始化为0来表示初始状态。例如，在计算斐波那契数列的动态规划解法中，dp[0] = dp[1] = 1，其他元素初始为0。代码如下：

     int dp[100];
     memset(dp, 0, sizeof(dp));
     dp[0] = dp[1] = 1;
     

   • 初始化数组为 - 1：
     • 有时候将数组初始化为 - 1可以表示一种特殊状态。比如在图的遍历问题中，如果使用一个数组来记录节点是否被访问过，初始化为 - 1可以表示未访问状态。而0或者其他正值可能会与实际的节点编号或者其他有意义的值冲突。代码如下：

     int visited[100];
     memset(visited, -1, sizeof(visited));
     

   • 初始化字符数组为空字符（‘\0’）：
     • 当处理字符串相关的问题时，经常需要将字符数组初始化为空字符串。在C/C++中，空字符串实际上就是一个以空字符'\0'结尾的字符数组。可以使用memset将字符数组的所有元素设置为'\0'来达到初始化空字符串的目的。例如：

     char str[100];
     memset(str, '\0', sizeof(str));
     

6. 时间复杂度

   • memset函数的时间复杂度是线性的，为$O(n)$，其中n是要填充的字节数。这是因为memset函数是逐个字节地对内存区域进行填充操作。在实际应用中，如果要填充的字节数非常大，它的执行时间会随着字节数的增加而线性增加。不过在大多数情况下，由于其操作简单直接，并且在初始化数组等操作中是一种高效的方式，所以在算法比赛中被广泛使用。但要注意正确使用它来填充合适的字节数，避免因错误填充导致的程序错误。