程序找茬：统计字符个数问题

【题目描述】

下面的程序意图在于统计字符串中字符1的个数，可惜有瑕疵：

#include<stdio.h>
#define maxn 10000000 + 10
int main() {
    char s[maxn];
    scanf("%s", s);
    int tot = 0;
    for(int i = 0; i < strlen(s); i++)
        if(s[i] == 1) tot++;
    printf("%d\n", tot);
}

该程序至少有3个问题，其中一个导致程序无法运行，另一个导致结果不正确，还有一个导致效率低下。你能找到它们并改正吗？

【题目来源】

刘汝佳《算法竞赛入门经典  第2版》第3章 思考题2　统计字符1的个数

【解析】

这段代码有下述几个问题：

1．数组定义位置错误：导致程序无法运行

这个问题在“数组定义在main函数内外的区别”一文中有详细的阐述。因为代码中定义的数组很大，定义在main函数之内会导致栈溢出错误。解决方法是把数组定义在main函之外。

2．字符和整数的比较错误：导致结果不正确

字符的实际值是ASCII值，字符'1'的ASCII值不是1，而是49。因此，if(s[i] == 1) 是错误的，应改为：if(s[i] == '1')。

3．重复调用函数：导致效率低下

将计算字符串长度的函数strlen(s)写在循环中，会导致函数的重复调用，即每进行一次循环条件判断，都要重新计算字符串s的长度，显然这样会造成效率低下。

解决的办法也很简单，就是在循环之前计算字符串的长度，将值存储在一个变量中。

我们在后续编程时应注意：如果函数的返回值是确定的，不要将其放在循环中，而应预先将其值存入变量。

4．头文件缺失：导致编译时警告

strlen函数是定义在<string.h>头文件中的，上述代码并未引用该头文件。

实际测试，编译时会出现警告，但也可以正常运行，所以这个问题不是个严重问题。

不过即便不影响运行，还是加上为好。

5．缺少返回语句：不是问题的问题

代码没有返回语句：

return 0;

严格来讲这个问题不能算作问题，因为C语言标准规定了如果main函数运行到结尾时没有遇到return语句，那么它会隐式地返回0。所以写不写都是一样，但不写的话一般会被认为对代码清晰性有影响（其实也影响不了啥清晰性，说白了就是大家都加，你不加你就是异类，让人看了心生违和感）。因而出于编程习俗方面的考虑，还是加上为好。

修改后的代码如下：

#include<stdio.h>
#include<string.h> //引用strlen函数的头文件
#define maxn 10000000 + 10
char s[maxn]; //将数组定义在main函数之外
int main() {
    scanf("%s", s);
    int len = strlen(s); //在循环前预先计算字符串长度
    int tot = 0;
    for(int i = 0; i < len; i++)
        if(s[i] == '1') tot++; //使用字符'1'而不是整数1
    printf("%d\n", tot);
    return 0; //最好加上返回语句
}