D. Mysterious Crime(思维+前缀+公共子串)

https://codeforces.com/problemset/problem/1043/D


思路:

首先模拟发现最后的答案就是全部行的公共子串的总数。

怎么维护?子串的总数存在递推关系。

长度为3的等于长度为2的总数+1.然后答案再累加上此时的新贡献

问题就变成了维护长度为2的公共子串。

而且多个行的判断换到一个公共母串上去。

先将每一行的一个数的前一个数存起来(相当于一个前驱数组),这样我们查找第i个数的时候就可以直接知道第i-1个数了。

然后O(n*m)枚举可以o1判当前的数字前面是否一样,一样说明存在长度为的2的公共子串。然后递推计数累加到答案。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=1e5+1000;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL pre[maxn];
LL a[20][maxn];
LL sum[maxn];
int main(void){
   LL n,m;n=read();m=read();
   for(LL i=1;i<=m;i++){
      for(LL j=1;j<=n;j++){
         pre[j]=read();
         a[i][pre[j]]=pre[j-1];
      }
   }
   LL ans=n;
   for(LL i=2;i<=n;i++){///枚举每一位上的数字
      bool flag=1;
      for(LL j=1;j<m;j++){
            if(a[j][pre[i]]!=a[m][pre[i]]){
                flag=0;
                break;
            }
      }
      if(flag==1){
         sum[i]=sum[i-1]+1;///如果是分开的公共子串前面sum[i-1]为0
         ans+=sum[i];
      }
   }
   printf("%lld\n",ans);
   return 0;
}

猜你喜欢

转载自blog.csdn.net/zstuyyyyccccbbbb/article/details/115335638
今日推荐