C/C++二维数组传参

C/C++里二维数组传参时需要指定每行有几个元素(第二维长度),具体有三种常见的写法:  

// 参数传int a[][]过不了类型检查
void foo1(int a[][2]) {
    a[1][1] = 10;
}

void foo2(int *a[2]) {
    a[1][1] = 10;
}

void foo3(int **a) {
    a[1][1] = 10;
    // 等价的寻址写法
    // *(*(a+1)+1) = 10;
}

这三种写法在传参的性能上没有区别,在64位机器上都是传了8 bytes的指针,可以在 Compiler Explorer 上在线查看对应的汇编代码:

foo1(int (*) [2]):
        push    rbp
        mov     rbp, rsp
        mov     QWORD PTR [rbp-8], rdi
        mov     rax, QWORD PTR [rbp-8]
        add     rax, 8
        mov     DWORD PTR [rax+4], 10
        nop
        pop     rbp
        ret
foo2(int**):
        push    rbp
        mov     rbp, rsp
        mov     QWORD PTR [rbp-8], rdi
        mov     rax, QWORD PTR [rbp-8]
        add     rax, 8
        mov     rax, QWORD PTR [rax]
        add     rax, 4
        mov     DWORD PTR [rax], 10
        nop
        pop     rbp
        ret
foo3(int**):
        push    rbp
        mov     rbp, rsp
        mov     QWORD PTR [rbp-8], rdi
        mov     rax, QWORD PTR [rbp-8]
        add     rax, 8
        mov     rax, QWORD PTR [rax]
        add     rax, 4
        mov     DWORD PTR [rax], 10
        nop
        pop     rbp
        ret

由于都传的是指针而非值拷贝,这三种写法都可以做到覆盖参数中的数组值,具体可以试试下面的main函数:

#include <cstdio>

// code for foo1,foo2,foo3 omitted

int main() {
    int a[2][2];
    foo1(a);
    // printf("%d\n", a[1][1]);
    printf("%d\n", *(*(a+1)+1));

    int *b[2] = {new int[2], new int[2]};
    foo2(b);
    // printf("%d\n", b[1][1]);
    printf("%d\n", *(*(b+1)+1));
    delete b[0];
    delete b[1];

    int **c;
    c = new int*[2];
    c[0] = new int[2];
    c[1] = new int[2];
    foo3(c);
    // printf("%d\n", c[1][1]);
    printf("%d\n", *(*(c+1)+1));
    delete c[0];
    delete c[1];
    delete c;
}

另外,腾讯云上的一篇 技术分享 提到了基于模版的写法,因为二维数组传参要指定第二维长度,使用模版可以方便复用。


版权声明:本文为max_kibble原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。