蓝桥杯练习-U型数字

传送门： https://nanti.jisuanke.com/t/20683

最近蒜头君喜欢上了U型数字，所谓U型数字，就是这个数字的每一位先严格单调递减，后严格单调递增。比如 212212 就是一个U型数字，但是 333333, 9898, 567567, 3131331313，就是不是U型数字。

现在蒜头君问你，[1,100000][1,100000] 有多少U型数字？

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

int main()
{
    int ans=0;
    for (int i=100; i<=100000; i++)
    {
        //cout<<"!"<<endl;
        int a[7]={0};
        int m=i;
        int k=0;
        while (m)
        {
            a[k++] = m%10;
            m/=10;
        }
        int sum=0;
        int flag=1;
        for (int j=k-1; j>0; j--)
        {
            if (a[j]==a[j-1])
            {
                sum=0;
                break;
            }
            if (a[j]<a[j-1])
            {
                if (flag==0)
                {
                    sum++;
                    flag=1;
                }
            }
            else if (a[j]>a[j-1])
            {
                if (flag==1)
                {
                    sum++;
                    flag=0;
                }
            }
        }
        if (sum==2&&a[k-1]>a[k-2])
        {
            ans++;
            //cout<<i<<endl;
        }

    }
    cout<<ans<<endl;
}