P1311 选择客栈

嗯这道题我的做法是典型的用空间换时间

用a[i][j]记录颜色为i的客栈出现的坐标，每次枚举当前客栈到前一个同颜色的客栈间是否有符合条件的客栈，如果有，就是当前客栈之前出现的所有同色客栈数量，如果没有，就是上一个同色客栈的答案数，最后从头到尾遍历一遍每个客栈的答案累加，输出，结束

但是就在我写这篇博客的时候，我突然意识到，对于a数组，当当前节点为j时，我最多只用到了同色的客栈数以及a[i][j-1]，那不就可以优化一维只记录上次同色客栈的答案数么？？？？？？？那不就不用用空间换时间了？？？？？

md我真是个智障。。

附上我无脑开的不知道浪费了多少空间的代码。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#include<deque>
#include<algorithm>
#define ll long long
using namespace std;
const ll oo=0x3f3f3f3f;
const ll N=200005;
const ll M=55;

ll n,k,p,sum;
ll ans[N],a[M][N],b[N];

ll Min(ll a,ll b){return a<b?a:b;}
ll Max(ll a,ll b){return a>b?a:b;}
ll Abs(ll a){return a>0?a:-a;}

ll get(){
    char zy=getchar();
    ll z=1,y=0;
    while(zy>'9'||zy<'0'){
        if(zy=='-') z=-1;
        zy=getchar();
    }
    while(zy>='0'&&zy<='9'){
        y=(y<<1)+(y<<3)+zy-'0';
        zy=getchar();
    }
    return z*y;
}

int main(){
    //freopen("P1311选择客栈.in","r",stdin);
    //freopen("P1311选择客栈.out","w",stdout);
    n=get();k=get();p=get();
    for(ll i=1;i<=n;i++){
        bool ok=false;
        ll c=get();b[i]=get();
        if(a[c][0]){
            for(ll j=a[c][a[c][0]];j<=i;j++){
                if(b[j]<=p){
                    ok=true;
                    break;
                }
            }
            if(ok) ans[i]=a[c][0];
            else ans[i]=ans[a[c][a[c][0]]];
        }
        a[c][++a[c][0]]=i;
    }
    for(ll i=1;i<=n;i++) sum+=ans[i];
    printf("%lld
",sum);
    return 0;
}