使用main函数的参数,实现一个简单的整数计算器,具有整数加,减,乘,除运算的功能。
首先,我们先来看一下main函数参数是什么。
main函数有两种形式:
1. int main(int argc, char *argv[ ])
2. int main(int argc, char **argv)
其中:
argc为main函数接收字符串的个数;
argv[]为char型指针数组,数组中各指针指向输入的命令字符串。
举个栗子:
这里
argv[0]指向字符串 “./test”
argv[1]指向字符串 “-s”
argv[2]指向 “10”
argv[3]指向 “20”
argc为4,(共有四个字符串命令)
现在看代码:
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
//提示用法
static void usage(const char* proc)
{
printf("usage: \r\n\t %s [-a -s -m -d] num1 num2 \n", proc);
}
//加法
int myadd(int x, int y)
{
return x + y;
}
//减法
int mysub(int x, int y)
{
return x - y;
}
//乘法
int mymul(int x, int y)
{
return x*y;
}
//除法
int mydiv(int x, int y)
{
if (y == 0) //除数为0,参数错误
{
printf("The denominator cannot be 0!\n");
return -1;
}
return x / y;
}
int main(int argc, char* argv[])
{
if (argc != 4) //如果用户输入的不是四个字符串( ./a.out -* num1 num2 )
{
usage(argv[0]); //调用函数打印计算器使用方法
return 1;
}
int(*funp[])(int, int) = { myadd, mysub, mymul, mydiv }; //函数指针数组
int opt = 0;
if (strcmp(argv[1], "-a") == 0)
{
opt = 0;
}
else if (strcmp(argv[1], "-s") == 0)
{
opt = 1;
}
else if (strcmp(argv[1], "-m") == 0)
{
opt = 2;
}
else if (strcmp(argv[1], "-d") == 0)
{
opt = 3;
}
else
{
usage(argv[0]);
return 2;
}
int num1 = atoi(argv[2]); //将字符串转为int型数
int num2 = atoi(argv[3]);
int res = (funp[opt])(num1, num2);
printf("result: is %d\n", res);
return 0;
}
运行结果:
加减乘除都运算正确,输入错误的命令也能打印出usage。