写一个函数,用户输入一个数判断是否是素数

//写一个函数,用户输入一个数判断是否是素数 
 function get(num) {
            if (num == 1 || num == 0) {
                return num + '不是质数'
            }
            for (var i = 2; i < num; i++) {
                if (num % i == 0) {
                    return num + '不是质数'
                }
            }
            return num + '是质数'
        }
        alert(get(Number(prompt('请输入一个数'))));