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

时间:2023-01-10 23:31:36

题目: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 ;
}