C++ 打印极值点下标（九度OJ 1126）

题目描述：

在一个整数数组上，对于下标为i的整数，如果它大于所有它相邻的整数，
或者小于所有它相邻的整数，则称为该整数为一个极值点，极值点的下标就是i。

输入：

每个案例的输入如下：

有2×n+1行输入：第一行是要处理的数组的个数n；
对其余2×n行，第一行是此数组的元素个数k(4<k<80)，第二行是k个整数，每两个整数之间用空格分隔。

输出：

每个案例输出为n行：每行对应于相应数组的所有极值点下标值，下标值之间用空格分隔。

样例输入：

3
10
10 12 12 11 11 12 23 24 12 12
15
12 12 122 112 222 211 222 221 76 36 31 234 256 76 76 
15
12 14 122 112 222 222 222 221 76 36 31 234 256 76 73

样例输出：

0 7
2 3 4 5 6 10 12
0 2 3 10 12 14

源代码：

#include <iostream>
using namespace std;
int main()
{
    int n,i;
    int num;               //数组元素个数
    int a[80];
    int flag;              //flag用于判断是否打印空格
    cin>>n;
    while(n>0)
    {
        cin>>num;
        flag=0;
        for(i=0;i<num;i++)
        {
            cin>>a[i];
        }
        if(a[0]!=a[1])     //首个极值点
        {
            cout<<"0";
            flag++;
        }
        for(i=1;i<num-1;i++)
        {
            if((a[i]>a[i-1]&&a[i]>a[i+1]) || (a[i]<a[i-1]&&a[i]<a[i+1]))      //中间的极值点
            {
                if(flag!=0)
                {
                    cout<<" "<<i;
                    flag++;
                }
                else
                {
                    cout<<i;
                    flag++;
                }
            }
        }
        if(a[num-2]!=a[num-1])     //最后一个极值点
        {
            if(flag!=0)
            {
                cout<<" "<<num-1;
                flag++;
            }
            else
            {
                cout<<num-1;
                flag++;
            }
        }
        cout<<endl;
        n--;
    }
    return 0;
}

程序截图：