P7324 [WC2021] 表达式求值
P7324 [WC2021] 表达式求值
闲话
WC2021 我只得了 20 分,三道题总共 20 分。我是下场了突然后知后觉这件事的,主要原因是我开了 C++11,然后 T1 T2 都没分了。在洛谷上测本来能拿银牌的。T1 的乱搞能拿 48,还挺高的。
幸亏咱们陕西省选不看冬令营成绩。幸亏是在省选前犯的这个错误。告诫后人和自己,写题前一定要看编译选项,否则只能后悔莫及。
T2 场上写的是不带问号的 \(O(n|E|)\) 和带问号 \(O(n|E|m^2)\) 的 70 分暴力。后者可以用 minmax 卷积优化掉一个 \(m\) 获得 85 分的好成绩。
思路
首先要会上面提到的带问号的暴力,不需要优化。
我在场上写的是这样子的:
int a[maxn][10],but[maxn][10],tmp[10];
int* solve(int l,const int &r,const int *a){
l++;
int *ans=s[l]=='('?solve(l,to[l],a):but[l];//这里的 to 是预处理的括号匹配位置,完全可以用把 l 设成全局变量的方法解决
l=s[l]=='('?to[l]+1:l+1;
while(l+1<r){
char op=s[l++];
int *res=s[l]=='('?solve(l,to[l],a):but[l];
l=s[l]=='('?to[l]+1:l+1;
switch(op){
case '<':{
memset(tmp,0,sizeof tmp);
for(int i=0;i<m;i++) for(int j=0;j<m;j++) if(a[i]<a[j]) tmp[i]=(tmp[i]+1ll*ans[i]*res[j])%mod;
else tmp[j]=(tmp[j]+1ll*ans[i]*res[j])%mod;
for(int i=0;i<m;i++) ans[i]=tmp[i];
break;
}
case '>':{
memset(tmp,0,sizeof tmp);
for(int i=0;i<m;i++) for(int j=0;j<m;j++) if(a[i]>a[j]) tmp[i]=(tmp[i]+1ll*ans[i]*res[j])%mod;
else tmp[j]=(tmp[j]+1ll*ans[i]*res[j])%mod;
for(int i=0;i<m;i++) ans[i]=tmp[i];
break;
}
case '?':{
memset(tmp,0,sizeof tmp);
for(int i=0;i<m;i++) for(int j=0;j<m;j++) tmp[i]=(tmp[i]+1ll*ans[i]*res[j])%mod,tmp[j]=(tmp[j]+1ll*ans[i]*res[j])%mod;
for(int i=0;i<m;i++) ans[i]=tmp[i];
break;
}
}
}
return ans;
}
inline void work(){
for(int i=0;i<m;i++) for(int j=1;j<=n;j++) a[j][i]=star::a[i][j-1];
for(int i=1;i<=n;i++){
for(int j=1;j<=len;j++) for(int k=0;k<m;k++) but[j][k]=(isdigit(s[j]) and s[j]-'0'==k);
int* res=solve(1,len,a[i]);
for(int j=0;j<m;j++)
ans=(ans+1ll*res[j]*a[i][j])%mod;
}
printf("%d\n",ans);
}
即把 \(n\) 组数分开来讨论,求每种数贡献的方案数,最后乘上值加起来。
那么问题就在于求每种数的方案数,这样的工作我们需要做 \(n\) 遍,非常不划算。
然后发现,对于大小关系相同的每组数,每种数贡献的方案数是相同的,改变的只是它们的值,我们没有必要每次都算一遍方案数。
发现,一个数在一组数中的相对大小关系的情况总数只有 \(2^m\) 种,即是否比该数大。
那么我们预处理出一个数的所有相对大小关系总数(这个部分是 \(O(2^m|E|)\) 的),然后对于 \(n\) 组数每个枚举 \(m\) 个数差分统计答案即可(差分得到该数的方案数。此部分是 \(O(nmlogm)\) 的)。
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cctype>
#include<cstring>
#include<cmath>
#include<utility>
using namespace std;
inline int read(){
int w=0,x=0;char c=getchar();
while(!isdigit(c))w|=c=='-',c=getchar();
while(isdigit(c))x=x*10+(c^48),c=getchar();
return w?-x:x;
}
namespace star
{
const int maxn=5e4+10,maxm=12,S=1050,mod=1e9+7;
int n,m,a[maxn][maxm],D,f[S],pos,b[maxm],ans;
char s[maxn];
pair<int,int> solve(){
int num=0;bool ok=true;char lst='>';
pair<int,int> ans(1,0),now,tmp;
while(lst!=')' and ++pos)
if(isdigit(s[pos])) num=s[pos]^48;
else if(s[pos]=='(') now=solve(),ok=false;
else ok and (now=make_pair((D>>num)&1,(D>>num)&1^1),0),
ok=true,tmp=make_pair(0,0),
lst!='>' and (tmp.second=(tmp.second+1ll*ans.second*now.second)%mod,tmp.first=(tmp.first+1ll*(ans.first+ans.second)*(now.first+now.second)-1ll*ans.second*now.second%mod+mod)%mod),
lst!='<' and (tmp.first=(tmp.first+1ll*ans.first*now.first)%mod, tmp.second=(tmp.second+1ll*(ans.first+ans.second)*(now.first+now.second)-1ll*ans.first*now.first%mod+mod)%mod),
ans=tmp,lst=s[pos],num=0;
return ans;
}
inline bool cmp(const int &x,const int &y){return a[pos][x]<a[pos][y];}
inline void work(){
n=read(),m=read();
for(int i=0;i<m;i++) for(int j=0;j<n;j++) a[j][i]=read();
scanf("%s",s+1);s[strlen(s+1)+1]=')';
for(D=0;D<(1<<m);D++) pos=0,f[D]=solve().second;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++) b[j]=j;
pos=i;sort(b,b+m,cmp);
for(int j=0,d=0;j<m;d|=(1<<b[j++]))
ans=(ans+1ll*(a[i][b[j]]-(j?a[i][b[j-1]]:0))*f[d])%mod;
}
printf("%d\n",ans);
}
}
signed main(){
star::work();
return 0;
}