定义函数统计素数

题目描述

小亮想知道任意指定的两个数字之间有多少个素数，请帮他编程实现以下功能：用户输入两个正整数，系统自动输出两个数之间素数的个数。

输入描述:

输入为一小一大两个正整数，范围为(1~2000)，用空格隔开。

输出描述:

针对输入的两个正整数，输出它们之间（包括这两个数）素数的个数并换行。

示例1

输入

复制1 20

1 20

输出

复制8

8

示例2

输入

复制100 200

100 200

输出

复制21

21

#include <stdio.h>
#include<iostream>
#include <math.h>
int fun(int a,int b)
{
    int count=0;
    if(a<2)a=2;
    for(int i=a;i<=b;i++)
    {
        int flag=1;
        for(int j=2;j<=sqrt(i);j++){if(i%j==0){flag=0;break;
                                              }
                                   }
        if(flag==1)count++;
    }
    return count;
}
int main()
{
    int a,b;
    scanf("%d%d",&a,&b);
    printf("%d",fun(a,b));
}