Processing math: 100%

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 ,这样 sf 就可以互相转移。

效率: 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;
}
复制代码

 

posted @   xjqxjq  阅读(162)  评论(0编辑  收藏  举报
努力加载评论中...
点击右上角即可分享
微信分享提示