我正在尝试编写将从stdin读取整数并打印其gcd的程序。如果两个整数都是素数,我打印"prime"
。在结束程序之前打印到stderr
- "DONE"
。当用户输入错误数据时,我要打印到stderr
- "ERROR"
。所以我写了这段代码:
#include "stdio.h"
#include "nd.h"
#include "nsd.h"
int main() {
int in1, in2;
int tmp;
while (1) {
tmp = scanf("%d %d", &in1, &in2);
if (tmp == EOF) {
fprintf(stderr, "DONE\n");
break;
}
if (tmp != 2) {
fprintf(stderr, "ERROR\n");
} else if ((nd(in1) == 1) && (nd(in2) == 1)) printf("prime\n");
// nd(int a) return 1 when a is prime
else printf("%d\n", gcd(in1, in2));
}
return 0;
}
我希望在"ERROR"
之后继续工作。但它不起作用,我尝试在continue;
之后添加fprintf(stderr, "ERROR\n");
,但它也不起作用。所以,我想:
- program run
5 10
5
1 3
prime
1 0.9
error
// not break here!
10 2
2
...
//EOF
DONE
- program stop
一切正常,除了"ERROR"
,我有这个:
- program run
5 0.9
ERROR
ERROR
ERROR
ERROR
...
//never stop
我理解在循环周期中它是正确的工作。我的问题是我需要改变什么来改变我所拥有的'到了我想要的东西'。
答案 0 :(得分:2)
scanf()
在尝试应对意外输入时遇到问题。而是使用fgets()
char buf[100];
if (fgets(buf, sizeof buf, stdin) == NULL) {
fprintf(stderr, "DONE\n");
break;
}
if (sscanf(buf, "%d%d", &in1, &in2) != 2) {
fprintf(stderr, "ERROR\n");
} else if ((nd(in1) == 1) && (nd(in2) == 1)) {
printf("prime\n");
} else {
printf("%d\n", gcd(in1, in2));
}
修改了在线上寻找额外文字的代码。
// if (sscanf(buf, "%d%d", &in1, &in2) != 2) {
// fprintf(stderr, "ERROR\n");
int n = 0;
if (sscanf(buf, "%d%d %n", &in1, &in2, &n) != 2 || buf[n]) {
fprintf(stderr, "ERROR\n");
} ...