使用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。