P3952 [NOIP2017 提高组] 时间复杂度-模拟+栈+字符串处理

[NOIP2017 提高组] 时间复杂度

题目背景

NOIP2017 提高组 D1T2

题目描述

小明正在学习一种新的编程语言 A++，刚学会循环语句的他激动地写了好多程序并 给出了他自己算出的时间复杂度，可他的编程老师实在不想一个一个检查小明的程序， 于是你的机会来啦！下面请你编写程序来判断小明对他的每个程序给出的时间复杂度是否正确。

A++语言的循环结构如下：

F i x y
    循环体
E

其中F i x y表示新建变量 $i$（变量 $i$ 不可与未被销毁的变量重名）并初始化为 $x$， 然后判断 $i$ 和 $y$ 的大小关系，若 $i$ 小于等于 $y$ 则进入循环，否则不进入。每次循环结束后 $i$ 都会被修改成 $i+1$，一旦 $i$ 大于 $y$ 终止循环。

$x$ 和 $y$ 可以是正整数（$x$ 和 $y$ 的大小关系不定）或变量 $n$。$n$ 是一个表示数据规模的变量，在时间复杂度计算中需保留该变量而不能将其视为常数，该数远大于 $100$。

E 表示循环体结束。循环体结束时，这个循环体新建的变量也被销毁。

注：本题中为了书写方便，在描述复杂度时，使用大写英文字母 $O$ 表示通常意义下 $\Theta$ 的概念。

输入格式

输入文件第一行一个正整数 $t$，表示有 $t$（$t\le 10$）个程序需要计算时间复杂度。 每个程序我们只需抽取其中 F i x y 和 E 即可计算时间复杂度。注意：循环结构允许嵌套。

接下来每个程序的第一行包含一个正整数 $L$ 和一个字符串，$L$ 代表程序行数，字符串表示这个程序的复杂度，O(1) 表示常数复杂度，O(n^w) 表示复杂度为 $n^w$，其中 $w$ 是一个小于 $100$ 的正整数，输入保证复杂度只有 O(1) 和 O(n^w) 两种类型。

接下来 $L$ 行代表程序中循环结构中的F i x y或者 E。 程序行若以F开头，表示进入一个循环，之后有空格分离的三个字符（串）i x y， 其中 $i$ 是一个小写字母（保证不为$n$），表示新建的变量名，$x$ 和 $y$ 可能是正整数或 $n$ ，已知若为正整数则一定小于 $100$。

程序行若以E开头，则表示循环体结束。

输出格式

输出文件共 $t$ 行，对应输入的 $t$ 个程序，每行输出 Yes 或 No 或者 ERR，若程序实际复杂度与输入给出的复杂度一致则输出 Yes，不一致则输出 No，若程序有语法错误（其中语法错误只有: ① F 和 E 不匹配 ②新建的变量与已经存在但未被销毁的变量重复两种情况），则输出 ERR。

注意：即使在程序不会执行的循环体中出现了语法错误也会编译错误，要输出 ERR。

样例 #1

样例输入 #1

8
2 O(1)
F i 1 1
E
2 O(n^1)
F x 1 n
E
1 O(1)
F x 1 n
4 O(n^2)
F x 5 n
F y 10 n
E
E
4 O(n^2)
F x 9 n
E
F y 2 n
E
4 O(n^1)
F x 9 n
F y n 4
E
E
4 O(1)
F y n 4
F x 9 n
E
E
4 O(n^2)
F x 1 n
F x 1 10
E
E

样例输出 #1

Yes
Yes
ERR
Yes
No
Yes
Yes
ERR

 #include <bits/stdc++.h>
using namespace std;
#define de(x) cout<<x<<" ";
#define Pu puts("");
#define sf(x) scanf("%d",&x);
typedef long long ll;
const int N=2e3+10,mod=100003;
int n,m,ans;
bool st[N];//记录变量是否出现过
stack<char>bian;//记录变量个数
stack<int>cnt;
int ERR;
int zhuan(char srr[]){
    int len=strlen(srr);
    if(len==1){
        if(srr[0]=='n') return 105;
    }
    return atoi(srr);
}
int main(){
    int T;cin>>T;
    char arr[N];
    int f[5];//记录输入中的时间复杂度

    char op,t;//输入F或者E操作；输入变量名称
    char srr[10];
    int l,r;
    while(T--){
        ERR=0;ans=1;
        f[0]=f[1]=0;
        while(bian.size()) bian.pop();
        while(cnt.size()) cnt.pop();
        memset(st,0,sizeof(st));

        cin>>n>>arr;
        if(arr[2]=='1') f[0]=1;//获取输入中的时间复杂度
         else{
            int st=4;
            while(arr[st]>='0'&&arr[st]<='9'){
                f[1]=f[1]*10+(arr[st]-'0');
                st++;
            }
        }
        int now=1;

        while(n--){
            cin>>op;
            if(op=='F'){
                ans=max(ans,now);//这里以连续个E为一个模块
                //因为可能最外层的for直接退出，而内层的for有复杂度
                //此时应该直接为1
                now=1;
                cin>>t;//防止多个变量重名
                if(st[t-'a']==1){
                    ERR=1;
                }
                st[t-'a']=1;
                bian.push(t);

                cin>>srr;
                l=zhuan(srr);
                cin>>srr;
                r=zhuan(srr);

                if(l<r&&r==105) cnt.push(2);//代表时间复杂度可以叠加
                else if(l<=r) cnt.push(1);
                else if(l>r) cnt.push(-1);
            }else{
                if(bian.size()==0) {ERR=1;continue;}//考虑E大于F的情况
                st[bian.top()-'a']=0;
                bian.pop();//这个变量已经可以用了
                int tmp=cnt.top();cnt.pop();//计算这连续个E得到的最终结果

                if(tmp==-1) {now=1;continue;}

                now*=tmp;
            }
        }
        ans=max(ans,now);
       
        if(bian.size()!=0) ERR=1;//考虑F大于E的情况
        if(ERR==1){
            printf("ERR\n");
        }else{
            if(f[0]==1){
                if(ans==1) printf("Yes\n");
                else printf("No\n");
            }else{
                int ok;
                for(int i=1;pow(2,i)<=ans;i++){
                    if(pow(2,i)==ans){
                        ok=i;
                        break;
                    }
                }
                if(ok==f[1]) printf("Yes\n");
                else printf("No\n");
            }
        }

    }
}

下面是Hack数据：
很多题解应该都过不了！！！