bzoj 5020(洛谷4546) [THUWC 2017]在美妙的数学王国中畅游——LCT+泰勒展开

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=5020

　　　https://www.luogu.org/problemnew/show/P4546

如果保证 x=1 ，则可以用 LCT 维护每个点的函数值。不然的话就用 LCT 拿出那条链，dfs 一下 splay 现算。可以得 60 分。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define db double
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
const int N=1e5+;
int n,m,fa[N],c[N][],sta[N],top; db sm[N],vl[N]; bool fx,rev[N];
struct Node{
  int f;db a,b;
  db cal(db x)
  {
    if(f==)return sin(a*x+b);
    if(f==)return exp(a*x+b);
    if(f==)return a*x+b;
  }
}a[N];
bool isroot(int cr){return c[fa[cr]][]!=cr&&c[fa[cr]][]!=cr;}
void pshp(int cr){sm[cr]=sm[c[cr][]]+sm[c[cr][]]+vl[cr];}
void Rev(int cr){if(rev[cr]){rev[cr]=;rev[c[cr][]]^=;rev[c[cr][]]^=;swap(c[cr][],c[cr][]);}}
void rotate(int x)
{
  int y=fa[x],z=fa[y],d=(x==c[y][]);
  if(!isroot(y))c[z][y==c[z][]]=x;//!isroot(y) not !isroot(z) !!
  fa[x]=z;
  fa[y]=x; fa[c[x][!d]]=y;
  c[y][d]=c[x][!d]; c[x][!d]=y;
  if(fx)pshp(y),pshp(x);
}
void splay(int x)
{
  sta[top=]=x;
  for(int cr=x;!isroot(cr);cr=fa[cr])sta[++top]=fa[cr];
  for(int i=top;i;i--)Rev(sta[i]);
  int y,z;
  while(!isroot(x))
    {
      y=fa[x];z=fa[y];
      if(!isroot(y))
    ((x==c[y][])^(y==c[z][]))?rotate(x):rotate(y);
      rotate(x);
    }
}
void access(int cr)
{
  for(int t=;cr;splay(cr),c[cr][]=t,pshp(cr),t=cr,cr=fa[cr]);//pshp()!!
}
void mkrt(int cr)
{
  access(cr);splay(cr);rev[cr]^=;
}
void link(int x,int y)
{
  mkrt(x);fa[x]=y;
}
void cut(int x,int y)
{
  mkrt(x);access(y);splay(y);
  fa[x]=c[y][]=;if(fx)sm[y]-=vl[x];
}
void mdfy(int cr,int f,db a1,db b1)
{
  if(fx)splay(cr),sm[cr]-=vl[cr];
  a[cr].f=f;a[cr].a=a1;a[cr].b=b1;
  if(fx)vl[cr]=a[cr].cal(),sm[cr]+=vl[cr];
}
int fnd(int cr)//root of real tree
{
  while(!isroot(cr))cr=fa[cr];return cr;
}
db dfs(int cr,db x)
{
  db ret=a[cr].cal(x);
  if(c[cr][])ret+=dfs(c[cr][],x);
  if(c[cr][])ret+=dfs(c[cr][],x);
  return ret;
}
void print(db x)
{
  bool fx=;int t=;
  if(x<){fx=;while(x<)x*=,t++;}
  if(x>=){while(x>=)x/=,t++;}
  printf("%.8fe%c%03d\n",x,fx?'-':'+',t);
}
int main()
{
  n=rdn();m=rdn();char ch[];scanf("%s",ch);
  if(ch[]=='')fx=;
  for(int i=;i<=n;i++)
    {
      a[i].f=rdn();scanf("%lf%lf",&a[i].a,&a[i].b);
      if(fx)vl[i]=sm[i]=a[i].cal();
    }
  int u,v;db a,b,x;
  while(m--)
    {
      scanf("%s",ch);
      if(ch[]=='a'){u=rdn()+;v=rdn()+;link(u,v);}
      if(ch[]=='d'){u=rdn()+;v=rdn()+;cut(u,v);}
      if(ch[]=='m'){u=rdn()+;v=rdn();scanf("%lf%lf",&a,&b);mdfy(u,v,a,b);}
      if(ch[]=='t')
    {
      u=rdn()+;v=rdn()+;scanf("%lf",&x);
      mkrt(u);access(v);splay(v);
      if(fnd(u)!=v)puts("unreachable");
      else print(fx?sm[v]:dfs(v,x));
    }
    }
  return ;
}

关于本题的求导和泰勒展开可以看这个：https://www.cnblogs.com/zhoushuyu/p/8148732.html

似乎还可以麦克劳林展开：https://www.cnblogs.com/Troywar/p/8982707.html

似乎展开成 12 项的多项式就差不多了，然后可以用 LCT 维护 12 项的系数和，算的时候乘上 x^k 再除以阶乘就行。

那个 x₀ 好像取 0 就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define db double
#define ls c[cr][0]
#define rs c[cr][1]
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
using namespace std;
const int N=1e5+,M=;
int n,fa[N],c[N][],sta[N],top; bool rev[N]; db sm[N][M+],vl[N][M+],jc[N];
struct Node{
  int f;db a,b;
  Node(int f=,db a=,db b=):f(f),a(a),b(b) {}
  void cz(int cr)
  {
    if(f==)
      {
    db p[],lja=;
    p[]=sin(b);p[]=cos(b);p[]=-p[];p[]=-p[];
    for(int i=;i<M;i+=)
      for(int j=;j<;j++,lja*=a)sm[cr][i+j]=vl[cr][i+j]=p[j]*lja;
      }
    if(f==)
      {
    db ml=exp(b),lja=;
    for(int i=;i<M;i++,lja*=a)sm[cr][i]=vl[cr][i]=ml*lja;
      }
    if(f==)
      {
    sm[cr][]=vl[cr][]=b; sm[cr][]=vl[cr][]=a;
    for(int i=;i<M;i++)sm[cr][i]=vl[cr][i]=;
      }
  }
}w[N];
bool isrt(int cr){return c[fa[cr]][]!=cr&&c[fa[cr]][]!=cr;}
void pshp(int cr)
{
  for(int i=;i<M;i++)sm[cr][i]=sm[ls][i]+sm[rs][i]+vl[cr][i];
}
void Rev(int cr){if(rev[cr]){rev[cr]=;rev[ls]^=;rev[rs]^=;swap(ls,rs);}}
void rotate(int x)
{
  int y=fa[x],z=fa[y],d=(x==c[y][]);
  if(!isrt(y))c[z][y==c[z][]]=x;
  fa[x]=z;
  fa[y]=x; fa[c[x][!d]]=y;
  c[y][d]=c[x][!d]; c[x][!d]=y;
  pshp(y);pshp(x);
}
void splay(int x)
{
  sta[top=]=x;
  for(int cr=x;!isrt(cr);cr=fa[cr])sta[++top]=fa[cr];
  for(int i=top;i;i--)Rev(sta[i]);
  int y,z;
  while(!isrt(x))
    {
      y=fa[x];z=fa[y];
      if(!isrt(y))
    ((x==c[y][])^(y==c[z][]))?rotate(x):rotate(y);
      rotate(x);
    }
}
void access(int cr)
{
  for(int t=;cr;splay(cr),rs=t,pshp(cr),t=cr,cr=fa[cr]);//pshp()!!!
}
void mkrt(int cr)
{
  access(cr);splay(cr);rev[cr]^=;
}
void link(int x,int y)
{
  mkrt(x);fa[x]=y;
}
void cut(int x,int y)
{
  mkrt(x);access(y);splay(y);
  fa[x]=c[y][]=;pshp(y);
}
void mdfy(int cr,int f,db a,db b)
{
  splay(cr);
  w[cr]=Node(f,a,b);w[cr].cz(cr);
}
db cal(int cr,db x)
{
  db ret=,ml=;
  for(int i=;i<M;i++,ml*=x)
    ret+=sm[cr][i]*ml/jc[i];
  return ret;
}
int fnd(int cr)
{
  while(!isrt(cr))cr=fa[cr];return cr;
}
void print(db x)
{
  bool fx=;int t=;
  if(x<){fx=;while(x<)x*=,t++;}
  if(x>=){while(x>=)x/=,t++;}
  printf("%.8fe%c%03d\n",x,fx?'-':'+',t);
}
int main()
{
  jc[]=;for(int i=;i<M;i++)jc[i]=jc[i-]*i;
  n=rdn();int m=rdn();char ch[];scanf("%s",ch);
  for(int i=;i<=n;i++)
    w[i].f=rdn(),scanf("%lf%lf",&w[i].a,&w[i].b),w[i].cz(i),pshp(i);//pshp for sm[]
  int u,v;db a,b;
  while(m--)
    {
      scanf("%s",ch);
      if(ch[]=='a'){u=rdn()+;v=rdn()+;link(u,v);}
      if(ch[]=='d'){u=rdn()+;v=rdn()+;cut(u,v);}
      if(ch[]=='m'){u=rdn()+;v=rdn();scanf("%lf%lf",&a,&b);mdfy(u,v,a,b);}
      if(ch[]=='t')
    {
      u=rdn()+;v=rdn()+;scanf("%lf",&a);
      mkrt(u);access(v);splay(v);
      if(fnd(u)!=v)puts("unreachable");
      else print(cal(v,a));
    }
    }
  return ;
}