【洛谷】3月月赛T3-Derivation-表达式树&拓展欧拉定理

传送门:luogu5240

开这道题果然是个错误的决定(debug了2h…)

---

题解

表达式树

树上每个节点维护原函数值和导函数值(求导需要维护原函数值)，用扩展欧拉定理展开$\log P$层

复杂度$O(T|S|{\log }^{2}P)$

注意：

1. 拓展欧拉定理下拓展出来的层在数的运算时一定注意$<\phi (p)$的值是原值，对于$x\ge \phi (p)$应取$x\%\phi (p)+\phi (p)$（模运算优化的时候注意一下）
2. 因为$x\le 2\phi (p)$，所以相加也可能会爆$\text{long long}$！
3. 一个细节：求导时会将指数-1，对于$<\phi (p)$和$>\phi (p)$的数-1在欧拉定理下性质不变(依旧满足$<\phi (p)$或$>\phi (p)$)。而当指数为$\phi (p)$时，应当转成$2\phi (p)-1$(不过代码里没有特判，数据没卡QWQ)
4. 用拓展欧拉定理展开需要多展开几层模为1的

---

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=10005,mod=998244353;

int tk,rt,n,v,ls[N],rs[N],val[N],md[50],cot,tot;
struct P{int f[50],df[50];}nw;
char op[N],s[N];

inline int phi(int x)
{
	int re=x,i;
	for(i=2;i*i<=x;++i) if(x%i==0){
		re=re/i*(i-1);
		for(;x%i==0;x/=i);
	}
    if(x>1) re=re/x*(x-1);
    return re;
}

inline int ad(ll x,ll y,int p){x+=y;return x>=p?x%p+p:x;}
inline int mul(ll x,ll y,int p){x*=y;return x>=p?x%p+p:x;}
inline int dc(ll x,ll y,int p){x--;return x<0?x+p:x;} 

inline int fp(int x,int y,int p)
{
	int re=1;
	for(;y;y>>=1,x=mul(x,x,p))
	 if(y&1) re=mul(re,x,p);
	return re;
}

inline bool jud(char c){return (c=='+')||(c=='*')||(c=='^');}

int build(int l,int r)
{
	if(s[l]=='(' && s[r]==')') l++,r--;
	int i=l,j=r,lv=0,rv=0,x=-1,u=++tot;
	for(;i<=j;){
		if(s[i]=='(') lv++;else if(s[i]==')') lv--;
		if(jud(s[i])&&(!lv)) {x=i;break;}i++;
		if(s[j]=='(') rv++;else if(s[j]==')') rv--;
		if(jud(s[j])&&(!rv)) {x=j;break;}j--;
	}
	if(x!=-1){op[u]=s[x];ls[u]=build(l,x-1);rs[u]=build(x+1,r);}
	else{
		if(s[r]=='x') op[u]='x',val[u]=1;
		else{
			for(x=0,i=l;i<=r;++i) x=x*10+(s[i]-'0');
			op[u]='C';val[u]=x;
		}
		ls[u]=rs[u]=0;//多组数据 
	}
	return u;
}

P dfs(int x)
{
	P re;int i,a,b;
	if((!ls[x])&&(!rs[x])){
		a=(op[x]=='x')?v:val[x],b=(op[x]=='x')?1:0;
		for(i=0;i<=cot;++i){
			re.f[i]=ad(a,0,md[i]);
			re.df[i]=ad(b,0,md[i]);
		}
	}else{
		P l=dfs(ls[x]),r=dfs(rs[x]);
		if(op[x]=='+'){
			for(i=0;i<=cot;++i){
				re.f[i]=ad(l.f[i],r.f[i],md[i]);
				re.df[i]=ad(l.df[i],r.df[i],md[i]);
			}
		}else if(op[x]=='*'){
		    for(i=0;i<=cot;++i){
		    	re.f[i]=mul(l.f[i],r.f[i],md[i]);
		    	re.df[i]=ad(mul(l.f[i],r.df[i],md[i]),mul(l.df[i],r.f[i],md[i]),md[i]);
		    }
		}else{
			re.f[cot]=re.df[cot]=0;
			for(i=cot-1;i>=0;--i){
		        re.f[i]=fp(l.f[i],r.f[i+1],md[i]);
		        re.df[i]=mul(mul(r.f[i],fp(l.f[i],dc(r.f[i+1],md[i+1]-1,md[i+1]),md[i]),md[i]),l.df[i],md[i]);
			}
		}
	}
	return re;
}

int main(){
	int i;md[0]=mod;
	for(i=phi(mod);i>1;i=phi(i)) md[++cot]=i;
	md[++cot]=1;md[++cot]=1;
	for(scanf("%d",&tk);tk;--tk){
		s[1]='(';
		scanf("%s",s+2);n=strlen(s+1);
		s[++n]=')';s[n+1]='\0';
		tot=0;rt=build(1,n);
		for(int i=0;i<2;++i){
	        scanf("%d",&v);
	        printf("%d",dfs(rt).df[0]%mod);
	        if(!i) putchar(' ');
			else if(tk>1) putchar('\n'); 
		}
	}
	return 0;
}