小美的代金券要过期啦（美团2021校招笔试+模拟栈）

思路

用栈模拟

代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[510],st[510];
int main()
{
    int n;
    scanf("%d",&n);
    int k=1;
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    st[k]=a[1];
    int ans=0;
    for(int i=2; i<=n; i++)
    {
        if(a[i]==st[k])
        {
            st[k]++;
            ans++;
            while(st[k]==st[k-1]&&k-1>=1)
            {
                st[k-1]++;
                k--;
                ans++;
            }
        }
        else
            st[++k]=a[i];
    }
    printf("%d\n",ans);
    return 0;
}