LOJ 3089 「BJOI2019」奥术神杖——AC自动机DP+0/1分数规划

题目：https://loj.ac/problem/3089

没想到把根号之类的求对数变成算数平均值。写了个只能得15分的暴力。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define db double
using namespace std;
const int N=,K=; const db eps=1e-;
int n,m,tot=,c[N][K],fl[N]; db ans;
int hd[N],xnt,to[N<<],nxt[N<<],vl[N],sm[N];
int l[N],v[N],dy[N],q[N]; db vp[N][N];
char s[N],ch[N],prn[N];
db pw(db x,int k)
{db ret=;while(k){if(k&)ret*=x;x*=x;k>>=;}return ret;}
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
int ins(int len)
{
  int cr=;
  for(int i=;i<=len;i++)
    {
      int w=ch[i]-'';
      if(!c[cr][w])c[cr][w]=++tot;
      cr=c[cr][w];
    }
  return cr;
}
void get_fl()
{
  int he=,tl=;
  for(int i=;i<;i++)
    if(c[][i])q[++tl]=c[][i],fl[c[][i]]=;
    else c[][i]=;
  while(he<tl)
    {
      int k=q[++he],pr=fl[k];
      for(int i=;i<;i++)
    if(c[k][i])q[++tl]=c[k][i],fl[c[k][i]]=c[pr][i];
    else c[k][i]=c[pr][i];
    }
  for(int i=;i<=tot;i++)add(fl[i],i);
}
void m_dfs(int cr)
{
  sm[cr]=vl[cr];
  for(int i=hd[cr],v;i;i=nxt[i])
    m_dfs(v=to[i]), sm[cr]+=sm[v];
}
db fnd(int ct,db tp)
{
  db l=,r=tp,ret=;
  while(r-l>eps)
    {
      db mid=(l+r)/;
      if(pw(mid,ct)<=tp)ret=mid,l=mid+eps;
      else r=mid-eps;
    }
  return ret;
}
void dfs(int cr,int nw)
{
  if(cr>n)
    {
      m_dfs(); db tp=; int ct=;
      for(int i=;i<=m;i++)
    { tp*=vp[i][sm[dy[i]]]; ct+=sm[dy[i]];}
      db d=fnd(ct,tp);
      if(d>ans)
    { ans=d;for(int i=;i<=n;i++)prn[i]=s[i];}
      return;
    }
  if(s[cr]!='.')
    {
      nw=c[nw][s[cr]-'']; vl[nw]++;
      dfs(cr+,nw); vl[nw]--; return;
    }
  int ynw=nw;
  for(int i=;i<;i++)
    {
      s[cr]=i+''; nw=c[ynw][i];
      vl[nw]++; dfs(cr+,nw); vl[nw]--;
    }
  s[cr]='.';////
}
void solve1()
{
  for(int i=;i<=m;i++)
    {
      vp[i][]=;//[0]!!
      for(int j=,lm=n-l[i]+;j<=lm;j++)
    vp[i][j]=vp[i][j-]*v[i];
    }
  dfs(,);
  for(int i=;i<=n;i++)putchar(prn[i]); puts("");
}
namespace S2{
  bool en[N],dp[N][N]; int pr[N][N],pw[N][N];
  void dfs(int cr)
  {
    for(int i=hd[cr],v;i;i=nxt[i])
      en[v=to[i]]|=en[cr], dfs(v);
  }
  void solve()
  {
    for(int i=;i<=m;i++)en[dy[i]]=;
    m_dfs(); dp[][]=; int ni=,nj;
    for(int i=;i<=n&&!ni;i++)
      for(int j=;j<=tot;j++)
    if(dp[i][j])
      {
        if(en[j]){ni=i;nj=j;break;}
        if(s[i]=='.')
          {
        for(int w=;w<;w++)
          if(!dp[i+][c[j][w]])
            { pr[i+][c[j][w]]=j;
              pw[i+][c[j][w]]=w;
              dp[i+][c[j][w]]=;}
          }
        else
          {
        int w=s[i]-'';
        for(int j=;j<=tot;j++)
          if(!dp[i+][c[j][w]])
            { pr[i+][c[j][w]]=j;
              pw[i+][c[j][w]]=w;
              dp[i+][c[j][w]]=;}
          }
      }
    for(int i=ni;i>;i--)
      s[i-]=pw[i][nj]='', nj=pr[i][nj];
    for(int i=;i<=n;i++)
      if(s[i]=='.')putchar('');
      else putchar(s[i]);
    puts("");
  }
}
int main()
{
  scanf("%d%d",&n,&m); scanf("%s",s+);
  int ct=; for(int i=;i<=n;i++) ct+=(s[i]=='.');
  bool fg=;
  for(int i=;i<=m;i++)
    {
      scanf("%s",ch+); scanf("%d",&v[i]);
      l[i]=strlen(ch+); dy[i]=ins(l[i]);
      if(i>&&v[i]!=v[i-])fg=;
    }
  get_fl();
  if(n<=||ct<=)solve1(); else if(!fg)S2::solve();
  return ;
}

\( log(a^x) = x*log(a) \) ， \( log(a*b) = log(a)+log(b) \) 。和乘法之类有关的应该考虑一下 log 。

求平均值就用 0/1 分数规划。

不要每次合法的时候都把方案存一遍。二分完再做一次，得到方案。自己用 pr[ ][ ] 和 pw[ ][ ] 记录 dp[ i ][ j ] 是从 dp[ i-1 ] 的哪里转移来的、填了什么值。

二分的 r 的范围是 max( log(v) ) 。

m 个串里没出现过的字符都是等价的。记录一下 “出现过的字符” ， 如果有没出现过的就选一个加入字符集，然后用那个字符集来转移即可。

这样就能支持 1e-8 的精度了。不过 1e-4 就够了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define db double
using namespace std;
db Mx(db a,db b){return a>b?a:b;}
const int N=,K=; const db INF=3e7,eps=1e-;
int n,m,tot=,c[N][K],fl[N],q[N];
int ct[N],pr[N][N],pw[N][N]; db v[N],dp[N][N];
char s[N],ch[N],prn[N];
int tw[K],top; bool vis[K];
db get_lg(int d)
{
  db l=,r=,ret=;//ret=0
  while(r-l>eps)
    {
      db mid=(l+r)/;
      if(pow(,mid)<=d)ret=mid,l=mid+eps;
      else r=mid-eps;
    }
  return ret;
}
void ins(int len,db d)
{
  int cr=;
  for(int i=;i<=len;i++)
    {
      int w=ch[i]-'';
      if(!vis[w])vis[w]=,tw[++top]=w;
      if(!c[cr][w])c[cr][w]=++tot;
      cr=c[cr][w];
    }
  v[cr]=d; ct[cr]=;
}
void get_fl()
{
  int he=,tl=;
  for(int i=,j;i<;i++)
    if((j=c[][i]))q[++tl]=j,fl[j]=;
    else c[][i]=;
  while(he<tl)
    {
      int k=q[++he],pr=fl[k];
      v[k]+=v[pr]; ct[k]+=ct[pr];
      for(int i=,j;i<;i++)
    if((j=c[k][i]))q[++tl]=j,fl[j]=c[pr][i];
    else c[k][i]=c[pr][i];
    }
}
int chk(db L)
{
  for(int i=;i<=n;i++)
    for(int j=;j<=tot;j++)dp[i][j]=-INF;
  dp[][]=;
  for(int i=;i<n;i++)
    {
      for(int j=;j<=tot;j++)
    if(dp[i][j]>-INF)
      {
        if(s[i+]!='.')
          {
        int w=s[i+]-'',tj=c[j][w];
        db d=dp[i][j]+v[tj]-L*ct[tj];
        if(d>dp[i+][tj])
          {
            dp[i+][tj]=d;pr[i+][tj]=j;pw[i+][tj]=w;
          }
        continue;
          }
        for(int k=;k<=top;k++)
          {
        int w=tw[k],tj=c[j][w];
        db d=dp[i][j]+v[tj]-L*ct[tj];
        if(d>dp[i+][tj])
          {
            dp[i+][tj]=d;pr[i+][tj]=j;
            pw[i+][tj]=w;
          }
          }
      }
    }
  int j=;
  for(int i=;i<=tot;i++)if(dp[n][i]>dp[n][j])j=i;
  return j;
}
int main()
{
  scanf("%d%d",&n,&m);scanf("%s",s+);
  db d,l=,r=,ans;
  for(int i=;i<=m;i++)
    {
      scanf("%s%lf",ch+,&d); d=get_lg(d);
      r=Mx(r,d);//Mx not sm
      int len=strlen(ch+); ins(len,d);
    }
  get_fl();
  for(int i=;i<;i++)if(!vis[i]){tw[++top]=i;break;}
  while(r-l>eps)
    {
      db mid=(l+r)/;
      if(dp[n][chk(mid)]>)
    //>0 not >=0 for almost always can =0
    { ans=mid; l=mid+eps;}
      else r=mid-eps;
    }
  int j=chk(ans);
  for(int i=n;i;i--)
    prn[i]=pw[i][j]+'', j=pr[i][j];
  printf("%s\n",prn+);
  return ;
}