Gym - 101246A Bencoding【模拟】

题目链接：https://vjudge.net/problem/Gym-101246A
题意：有三种形式的变量，分别为 整数：i数字e，字符串：字符串长度n:n个字符，列表：l列表内容e，字典：d字典内容e，注意整数中间不能出现其他的字符，也不能有前导零，也不能为负数，列表里面可以出现列表也可以出现整数和字符，字典里面的key可以重复，字典和列表都可以充当key和value，首先判断给出的字符串s是否合法（这个字符串只能表示一种类型），如果不合法请你找出s最长的前缀，你可以往前缀后面添加任意的字符，但总长度不能超过m，使得这个字符串合法，并输出这个这个前缀的最后一位，如果合法输出ok。
解析：由于这个字符串只能表示一种类型，所以通过第一个字符我们就可以知道这个字符串所表示的类型，接下来根据不同的类型进行判断，对于m每次都减去最少需要的添加的长度，然后慢慢模拟就可以了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <iostream>
#include <vector>
#include <queue>
#include <set>
using namespace std;
const int maxn = 1e6+100;
const int inf = 0x7fffffff;
char a[maxn];
int len,ans;
long long m;
void sloveDict(int &i);
void sloveInt(int &i);
void sloveList(int &i);
void sloveString(int &i);

void sloveInt(int &i)
{
    m-=3;
    if(m<0)
        ans = min(i,ans);
    int num = -1;
    while(ans==inf && a[++i]!='e')
    {
        if(num!=0 && a[i]>='0' && a[i]<='9')
        {
            if(num==-1)
                num = a[i]-'0';
            else
                m--;
            if(m<0)
                ans = min(ans,i);
        }
        else
            ans = min(ans,i);
    }
    if(num==-1)
        ans = min(ans,i);
}
void sloveString(int &i)
{
    long long tlen = a[i]-'0';
    m -= tlen+2;
    if(m<0)
        ans = min(ans,i);
    while(ans==inf && a[++i]!=':')
    {
        if(tlen!=0 && a[i]>='0' && a[i]<='9')
        {
            m -= (tlen*10+a[i]-'0')+1 - tlen;
            tlen = tlen*10+a[i]-'0';
            if(m<0)
                ans = min(ans,i);
        }
        else
            ans = min(ans,i);
    }
    while(tlen&&i<len)
        i++,tlen--;
    if(i==len)
        ans = min(ans,i);
}
void sloveList(int &i)
{
    m -= 2;
    if(m<0)
        ans = min(ans,i);
    while(ans==inf&&a[++i]!='e')
    {
        if(a[i]=='l')
            sloveList(i);
        else if(a[i]=='d')
            sloveDict(i);
        else if(a[i]=='i')
            sloveInt(i);
        else if(a[i]<='9' && a[i]>='0')
            sloveString(i);
        else
            ans = min(i,ans);
    }
}
void sloveDict(int &i)
{
    m-=2;
    if(m<0)
        ans = min(i,ans);
    int num = 0;
    while(ans==inf && a[++i]!='e')
    {
        num^=1;
        if(num)
            m -= 2;
        else
            m += 2;
        if(a[i]=='l')
            sloveList(i);
        else if(a[i]=='d')
            sloveDict(i);
        else if(a[i]=='i')
            sloveInt(i);
        else if(a[i]<='9' && a[i]>='0')
            sloveString(i);
        else
            ans = min(i,ans);
    }
    if(num)
        ans = min(i,ans);
}
int main()
{
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    scanf("%I64d %s",&m,a);
    len = strlen(a);
    int i = 0;
    ans = inf;
    if(a[0]=='l')
        sloveList(i);
    else if(a[0]=='d')
        sloveDict(i);
    else if(a[0]=='i')
        sloveInt(i);
    else if(a[0]<='9' && a[0]>='0')
        sloveString(i);
    else
        ans = min(0,ans);
    i++;
    if(i<len)
        ans = min(i,ans);
    if(ans!=inf)
        printf("Error at position %d!\n",ans);
    else
        puts("ok");
    return 0;
}