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 ;
}
上一篇:poj2728 Desert King【最优比率生成树】【Prim】【0/1分数规划】


下一篇:在Windows系统上一批可以下载但是需要经过编译再安装的第三方的直接编译后的版本(UCI页面)