codeforces 722 B. Verse Pattern

题目链接：传送门
题目大意是给n个字符串，和对应n个数字
要求每个字符串里面的元音字母个数要与对应的数字相等，相等输出yes，否则输出no
水题，遍历一遍就可以，但是元音字母里面还要加一个，不能不仔细看题呀…
代码如下：

#include <bits/stdc++.h>
using namespace std;
#define N 50010
#define mod 100000
#define inf 2000000000000000ll
int p[N];
char a[N];
int main(){
    int i,j,n,m;
   scanf("%d",&n);
        for(i=1;i<=n;i++)
            cin>>p[i];
            int flag=1;
           getchar();
    for(i=1;i<=n;i++){
            gets(a);
          int len=strlen(a);
            int x=0;
            if(flag==0) continue;
            for(j=0;j<len;j++)
                if(a[j]=='a'||a[j]=='e'||a[j]=='i'||a[j]=='u'||a[j]=='o'||a[j]=='y')
                x++;
            if(x!=p[i]) {
                flag=0;
            }
        }
            if(flag)
                cout<<"YES"<<endl;
            else cout<<"NO"<<endl;
    return 0;
}