SGU189 Perl-like Substr

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;

int n,m,tot;
struct String {
	string nam,text;
}s[N];

void Remove_Space(string &s) {        // 移除s中的空格、分号和换行符
	string sx="";
	int t=0;
	for (int i=0;i<s.length();i++) {
		if (s[i]==';') break;
		if (s[i]=='\"') t^=1;
		if (t||s[i]!=' ') sx+=s[i];
	}
	s=sx;
}

int Get_Num(string &s,int l,int r) {  // 读取出s[l,r]中的数字 
	int x=0,f=1;
	for (int i=l;i<=r;i++) {
		if (s[i]=='-') f=-1;
		else x=x*10+s[i]-'0';
	}
	return x*f;
}

int Get_Order(string &sx) {           // 在s[i]中查找名字为sx的字符串的下标
	for (int i=1;i<=tot;i++) {
		if (s[i].nam==sx) return i;
	}
	return 0;
}

string Substring(string &sx,int l,int r) { // 返回sx中substr函数对应的字符串，l、r分别为左右括号的下标 
	int a[4],cnt=0;
	for (int i=l;i<=r;i++) {
		if (sx[i]==',') a[++cnt]=i;
	}
	int l1=l+2,r1=a[1]-1,len1=r1-l1+1;
	string sy=sx.substr(l1,len1);
	int num=Get_Order(sy);
	if (cnt==1) {
		int beg=Get_Num(sx,a[1]+1,r-1);
		if (beg<0) beg=s[num].text.length()+beg;
		return s[num].text.substr(beg);
	}
	else if (cnt==2) {
		int beg=Get_Num(sx,a[1]+1,a[2]-1);
		if (beg<0) beg=s[num].text.length()+beg;
		int len=Get_Num(sx,a[2]+1,r-1);
		if (len<0) len=s[num].text.length()-beg+len;
		return s[num].text.substr(beg,len);
	}
}

void New_String(string &sx) { // 读入一个新字符串
	int a=sx.find('=');
	s[tot+1].nam=sx.substr(1,a-1);
	if (sx[a+1]=='$') {
		string ss=sx.substr(a+2,sx.length()-a-2);
		int num=Get_Order(ss);
		s[tot+1].text=s[num].text;
	}
	else if (sx[a+1]=='\"') {
		s[tot+1].text=sx.substr(a+2,sx.length()-a-3);
	}
	else if (sx[a+1]=='s') {
		s[tot+1].text=Substring(sx,a+7,sx.length()-1);
	}
	tot++;
}

string Equal(string &sx) { // 求出表达式对应的字符串值
	if (sx[0]=='$') {
		string ss=sx.substr(1,sx.length()-1);
		int num=Get_Order(ss);
		return s[num].text;
	}
	else if (sx[0]=='\"') {
		string ss="";
		for (int i=1;i<=sx.length()-2;i++)	ss+=sx[i];
		return ss;
	}
	else if (sx[0]=='s') {
		return Substring(sx,6,sx.length()-1);
	}
}

void Replace(string &sx,string &ss,int l,int r) { 将sx[l,r]中的内容替换成ss
	int a[4],cnt=0;
	for (int i=l;i<=r;i++) {
		if (sx[i]==',') a[++cnt]=i;
	}
	int l1=l+2,r1=a[1]-1,len1=r1-l1+1;
	string sy=sx.substr(l1,len1);
	int num=Get_Order(sy);
	if (cnt==1) {
		int beg=Get_Num(sx,a[1]+1,r-1);
		if (beg<0) beg=s[num].text.length()+beg;
		s[num].text.erase(beg);
		s[num].text=s[num].text.replace(beg,s[num].text.length()-beg,ss);
	}
	else if (cnt==2) {
		int beg=Get_Num(sx,a[1]+1,a[2]-1);
		if (beg<0) beg=s[num].text.length()+beg;
		int len=Get_Num(sx,a[2]+1,r-1);
		if (len<0) len=s[num].text.length()-beg+len;
		s[num].text=s[num].text.replace(beg,len,ss);
	}
}

int main() {
	scanf("%d%d\n",&n,&m); tot=0;
	s[0].text="";
	for (int i=1;i<=n+m;i++) {
		string sx; getline(cin,sx);
		Remove_Space(sx);
		if (sx[0]=='$') {
			int a=sx.find('=');
			string ss=sx.substr(1,a-1);
			int num=Get_Order(ss);
			if (num) {
				string sr=sx.substr(a+1,sx.length()-a-1);
				s[num].text=Equal(sr);
			}
			else New_String(sx);
		}
		
		else if (sx[0]=='p') {
			string sy=sx.substr(6,sx.length()-7);
			cout<<Equal(sy)<<endl;
		}
		else if (sx[0]=='s') {
			int a=sx.find('=');
			string ss=sx.substr(a+1,sx.length()-a-1);
			string sy=Equal(ss);
			Replace(sx,sy,6,a-1);
		}
	}
}