这个程序怎么写C++

描述
 用火柴棍拼成数字0～9，具体的拼法如下所示

                                        
 现给定一个任意三位数，请输出所需火柴棍的数量。
输入
 输入一行，为一个三位数
输出
 输出一行，为所需火柴棍的数量
样例输入

样例输出
 16

供参考：

#include <stdio.h>
int main()
{
             // 0 1 2 3 4 5 6 7 8 9
    int s[] = { 6,2,5,5,4,5,6,3,7,6 }, n, sum = 0;
    scanf("%d", &n);
    while (n)
    {
        sum += s[n % 10];
        n /= 10;
    }
    printf("%d", sum);
    return 0;
}