cf102012D. Rikka with Subsequences
题解
好难啊呜呜,怎么想的到。
三次方,那就相当于枚举可重复的三个相同的子序列的答案。
于是我们记 fi,j,k 表示第一个子序列最后一位为原序列中的第 i 个,第二个子序列为 j ,第三个为 k 的方案数。因此只有当 ai=aj=ak 才有值。
转移为 fi,j,k=Σfi′,j′,k′ ,其中 i′<i,j′<j,k′<k ,且 Mai′,ai=1 ,这样是 O(n6) 。
我们考虑前缀和优化,记 sj,k=Σfi′,j′,k′ ,并且 Mai′,aj=1 ,这样 s 和 f 就可以互相转移。
效率: O(n3) 。
代码
#include <bits/stdc++.h> using namespace std; const int N=202,P=1e9+7;; int n,a[N],ans,f[N][N],s[N][N],T; char m[N][N]; void work(){ scanf("%d",&n);ans=0; for (int i=1;i<=n;i++) scanf("%d",&a[i]); for (int i=1;i<=n;i++) scanf("%s",m[i]+1); for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) f[i][j]=s[i][j]=0; for (int i=1;i<=n;i++){ for (int j=1;j<=n;j++) for (int k=1;k<=n;k++) (f[j][k]+=((f[j-1][k]+f[j][k-1])%P+(P-f[j-1][k-1])%P)%P)%=P; for (int j=1;j<=n;j++) for (int k=1;k<=n;k++) if (m[a[i-1]][a[j]]=='1') (s[j][k]+=f[j-1][k-1])%=P; for (int j=1;j<=n;j++) for (int k=1;k<=n;k++){ if (a[i]==a[j] && a[i]==a[k]) f[j][k]=s[j][k]+1; else f[j][k]=0; (ans+=f[j][k])%=P; } } printf("%d\n",ans); } int main(){ for (scanf("%d",&T);T--;work()); return 0; }
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步