第7题-【字符串扫描】-注意处理头尾

/*
7. 读入一行文本，包含若干个单词（以空格间隔，％结尾）。将其中以 A 开头的
单词与以 N 结尾的单词，用头尾交换的办法予以置换。
*/

#include "iostream"
using namespace std;

#define MAX_SIZE 1001 + 1

int main()
{
char str[MAX_SIZE];
gets(str);

int i=0;
while(str[i])
{
if((str[i-1]==' ' && str[i]=='A') || str[0]=='A')
{
int k = i;
while(str[i] && str[i] != ' ')
{
i++;
}
if(str[i-1] == 'N')
{
char tmp = str[i-1];
str[i-1] = str[k];
str[k] = tmp;
}
if(!str[i])
{
break;
}
}
i++;
}

cout << str << endl;
return 0;
}