请编写函数fun，函数的功能是：判断字符串是否为回文？若是，函数返回1，主函数中输出：YES；否则返回0，主函数中输出NO。

#include <iostream>
using namespace std;
#define  N  80
int fun(char* str)
{
	int i = 1,j;
	j = strlen(str);
	for (i; i != '\0'; i++)
	{
		if (*(str + i) == str[j - 1 - i]) return 1;
		else return 0;
	}
}
int main()
{
	char  s[N];
	cout << "Enter a string: ";
	gets_s(s);
	cout << "\n\n";
	puts(s);
	if (fun(s)) cout << "  YES\n";
	else cout << "  NO\n";
}