codevs3945 完美拓印

3945 完美拓印

codevs月赛 第一场

时间限制: 1 s

空间限制: 256000 KB

题目等级 : 黄金 Gold

题目描述 Description

小Q获得了一个神奇的印章，这个印章宽n个单位长度，印章的其中三个棱都是直的，而另外一个方向上，对于每个单位宽度的部分，是一样直的，并且与反方向的棱平行，如下图所示。

小Q的印章上有一个不关于中心对称的图形（不一定是上图的Qrz），他现在要在一张地图上拓上印，地图上有一段个m单位长度、近似水平的边界线，但是放大到单位长度时还是有一定的高低差异，但对于单位宽度的部分，是一样直的，与水平轴线垂直，如下图所示。

小Q希望自己的印章一边的边缘能恰好地与边界线重合（不能部分重合、不能越过边界线），他现在只可以将印章旋转180度或者不旋转（这样印章可能存在有两边可以与边界线重合的情况），然后平移到适当的位置，问小Q有多少种可行的方案，两种方案不同被定义为两种方案用印章印出的图案互不重合。

 

 

 输入描述 Input Description

第一行两个正整数n和m，表示印章的宽度和截取边界线的长度。

第二行n个正整数，表示印章从左到右每个单位宽度对应的两条平行线之间的距离。

第三行m个整数，表示所截取边界线从左到右每个单位宽度对应的竖直方向上的坐标。

 

输出描述 Output Description

一个整数，表示可行方案的种类数。

 

样例输入 Sample Input

5 12

3 4 4 3 2

2 4 5 5 4 3 2 2 3 3 2 1

 

样例输出 Sample Output

3

 

数据范围及提示 Data Size & Hint

对于30%的数据，有n*m≤2*10⁷。

对于60%的数据，有n，m≤10⁵。

对于100%的数据，有n,m≤10⁶，所有数字的绝对值不超过10⁹。

 

【思路】

  字符串匹配。

  将一面的凹凸情况作为一面的特征，如果两面的凹凸状况相同则可以印章，则问题转化为求两种凹凸状况的匹配数目，可以用KMP算法求解。

  需要注意的是：

    1、   底边也可以匹配。

    2、   当n==1的时候应该特判（PT数组）。

【代码】

 

 1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 using namespace std; 5  6 const int maxn = 1000000+10; 7  8 int a[maxn],b[maxn],P[maxn],T[maxn],f[maxn]; 9 int n,m,ans;10 11 void get_Fail() {12     f[0]=f[1]=0;13     for(int i=1;i<n;i++) {14         int j=f[i];15         while(j && P[i]!=P[j]) j=f[j];16         f[i+1]=P[i]==P[j]?j+1:0;17     }18 }19 20 int KMP() {21     get_Fail();22     int j=0,cnt=0;23     for(int i=0;i<m;i++) {24         while(j && T[i]!=P[j]) j=f[j];25         if(T[i]==P[j]) j++;26         if(j==n) cnt++,j=f[j];27     }28     return cnt;29 }30 31 int main() {32     scanf("%d%d",&n,&m);33     for(int i=0;i<n;i++) scanf("%d",&a[i]);34     for(int i=0;i<m;i++) scanf("%d",&b[i]);35     if(n==1) {36         printf("%d\n",m*4);37         return 0;38     }39     40     n--; m--;41     for(int i=0;i<n;i++) P[i]=a[i+1]-a[i];42     for(int i=0;i<m;i++) T[i]=b[i+1]-b[i]; 43     ans += KMP();44     45     for(int i=0;i<n;i++) P[i]=a[n-i]-a[n-i-1];46     ans += KMP();47     48     for(int i=0;i<n;i++) P[i]=0;49     ans += 2*KMP();50     51     printf("%d\n",ans);52     return 0;53 }