P1737选择客栈

P1737选择客栈
无话可说，算难，但我自己做出来了

#include<stdio.h>
#include<string.h>
int n,k,p;
int a[200005][4];//a[i][0]代表颜色，a[i][1]表示价格，a[i][2]表示与它同色的有几个包括自己，a[i][3]表示上一个同色点的坐标（0就是之前没有了） 
int color[50];//表示上一个k色的位置 
int min=0;
int sum=0;
int main()
{
    int i;
    int temp;
    scanf("%d %d %d",&n,&k,&p);
    memset(a,0,sizeof(a));
    memset(color,0,sizeof(color));
    for(i=1;i<=n;i++)
    {
        scanf("%d %d",&a[i][0],&a[i][1]);
        if(color[a[i][0]]==0)
        {
            color[a[i][0]]=i;
            a[i][2]=1;
            a[i][3]=0;
        }
        else
        {
            a[i][2]=a[color[a[i][0]]][2]+1;
            a[i][3]=color[a[i][0]];
            color[a[i][0]]=i;
        }
        if(a[i][1]<=p)
        {
            min=i;
            sum+=a[i][2]-1;
        }
        else
        {
            temp=a[i][3];
            while(min<temp)
            temp=a[temp][3];
            sum+=a[temp][2];
        }
    }
    printf("%d",sum);
    return 0;
}