A+B problem 题解

先把一个单项式理解为：

符号，系数的绝对值，字母，指数。

为了方便操作，一口气读完整个字符串（数组），然后去扫描。

因为如果第一项为整数的话没有符号，判一判。

读入系数的绝对值像快读。

如果有 \(\texttt{^}\) 这个符号，读一下之后的指数。

由于只有三个字母，所以可以复制粘贴，不用写冗余的 \(\texttt{for}\) 循环。代码如下：

inline void change(polygon &a,char *s1){
	int l1=strlen(s1+1);
	for(int i=1;i<=l1;){
		int fl=1,x=0;
		d w1=nw,w2=nw,w3=nw;
		if(s1[i]=='+'||s1[i]=='-') fl=(s1[i]=='+'?1:-1),++i;
		while(isdigit(s1[i])) x=x*10+(s1[i]-'0'),++i;
		if(s1[i]=='x') w1.alpha=s1[i],++i;
		if(s1[i]=='^') ++i;
		while(isdigit(s1[i])) w1.digit=w1.digit*10+(s1[i]-'0'),++i; 
		w1.digit=!w1.digit?w1.alpha=='x':w1.digit;
		if(s1[i]=='y') w2.alpha=s1[i],++i;
		if(s1[i]=='^') ++i;
		while(isdigit(s1[i])) w2.digit=w2.digit*10+(s1[i]-'0'),++i; 
		w2.digit=!w2.digit?w2.alpha=='y':w2.digit;
		if(s1[i]=='z') w3.alpha=s1[i],++i;
		if(s1[i]=='^') ++i;
		while(isdigit(s1[i])) w3.digit=w3.digit*10+(s1[i]-'0'),++i; 
		w3.digit=!w3.digit?w3.alpha=='z':w3.digit;
		if(w1.alpha=='x'||w2.alpha=='y'||w3.alpha=='z') 
			a.x[{w1.digit,w2.digit,w3.digit}]+=fl*x;
		else a.q+=fl*x;
	}
}

然后考虑存储，约定一个结构体 \(str\) ，存储三个 \(\texttt{int}\) 变量 \(x,y,z\) 的质数。然后放进 \(\texttt{map}\) 里。映射项的系数。记得按照约定重载小于运算符。

定义一个多项式结构体，由一个 \(\texttt{map}\) 以及一个存储多项式常数的整型组成。

struct str{ 
	int x,y,z; 
	bool operator < (const str b) const{
		if(x+y+z!=b.x+b.y+b.z) return x+y+z>b.x+b.y+b.z;
		if(x!=b.x) return x>b.x;
		if(y!=b.y) return y>b.y;
		return z>b.z;
	}
};
struct polygon{	map<str,int> x; int q; }a,b,c;

表示 \(A,B,A+B.\)

合并同类项就扫一遍两个 \(\texttt{map}\) ，然后让新的 \(A+B\) 的每一项系数加上原来的系数。

inline void add(void){
	for(auto it:a.x) c.x[{it.first.x,it.first.y,it.first.z}]+=it.second;
	for(auto it:b.x) c.x[{it.first.x,it.first.y,it.first.z}]+=it.second;
	c.q=a.q+b.q;
}

然后按要求输出，这一步多加小心。\(\texttt{cbh}\) 可是很毒瘤的。

inline void output(polygon x){
	bool fl=0;
	for(auto it:x.x){
		if(!it.second) continue;
		if(it.second>0&&fl) printf("+");
		if(it.second==-1) printf("-");
		if(it.second!=1&&it.second!=-1) printf("%d",it.second);
		if(it.first.x==1) printf("x");
		if(it.first.x>1) printf("x^%d",it.first.x); 
		if(it.first.y==1) printf("y");
		if(it.first.y>1) printf("y^%d",it.first.y); 
		if(it.first.z==1) printf("z");
		if(it.first.z>1) printf("z^%d",it.first.z); 
		fl=1;
	}
	if(x.q){
		if(x.q>0&&fl) printf("+");
		printf("%d",x.q);
		fl=1;
	}
	if(!fl) printf("0");
	return puts(""),void();
}