求大神优化

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int flag[200010],n,k,p,c[200010],s[200010],sum,num;
int main()
{
    int m;
    scanf("%d%d%d",&n,&k,&p);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&c[i],&m);
        if(m>p)
        flag[i]=1;
    }
    for(int i=1;i<=n;i++)
    {
        if(flag[i]==0)
        {
            memset(s,0,sizeof(s));
            continue;
        }
        s[c[i]]++;
        sum+=s[c[i]]-1;
    }
    memset(s,0,sizeof(s));
    for(int i=1;i<=n;i++)
    {
        s[c[i]]++;
        num+=s[c[i]]-1;
    }
    printf("%d",num-sum);
    return 0;
}