URAL 1996 Cipher Message 3 (FFT + KMP)

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents by---cxlove

题意 ：给出两个串A ， B，每个串是若干个byte，A串的每个byte的最后一个bit是可以修改的。问最少修改多少，使得B串是A的一个子串。

2013年NEERC的题。。。。。。。感觉[buaa]sd0061教我做这题。

NEERC是毛子题，但是这套题感觉除了题面很难读之外，并不是很难。。。

目前为止，J题貌似全队都没看题，E题还在WA中，其它题都没啥问题。

做法：前7位都是不能修改的，所以要完全匹配，那么先按前7位KMP一下，记录匹配的位置。之后就是统计以这些位置开始，需要修改多少位，然后 取最小值。

所以提取出最后一个bit，成了两个01串，统计hamming distance。

记得若干年前就和队友讨论过这个问题，不过当时不是01串，觉得不可在n * m之内解决，然后 就放弃了。结果这个01串是可以 在nlgn解决 的。吓傻。。

两个01串a , b。将b反转之后，求一次卷积，便可以得到a串中以i为起始位置，与b进行匹配有多少个位置同为1。

那么接下来把a,b的01反转一下，再求一次卷积，就可以得到原串中为多少个位置同为0。就得到了hamming距离。

卷积求的是c[i + j] = a[i] * b[j]。由于我们将b进行了反转，所以a串中以i为起始位置的。

c[i + m - 1] = a[i + 0] * b[m - 0 - 1] + a[i + 1] * b[m - 1 - 1] + …… a[i + j] * b[m - j - 1] + …… a[i + m - 1] * b[m - (m - 1) - 1]。

如果a , b中同为1,乘积为1,否则为0,这样就统计出了有多少位同为1了。吓傻。。。太神了。

预处理出这个之后，再枚举之前KMP所得到的匹配位置 就可以了。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <vector>
#include <algorithm>
#include <queue>
#include <set>
using namespace std;
typedef long long LL;
typedef pair<int , int> pii;
const int N = 1000005;
//FFT copy from kuangbin
const double pi = acos (-1.0);
// Complex  z = a + b * i
struct Complex {
double a, b;
Complex(double _a=0.0,double _b=0.0):a(_a),b(_b){}
Complex operator + (const Complex &c) const {
return Complex(a + c.a , b + c.b);
}
Complex operator - (const Complex &c) const {
return Complex(a - c.a , b - c.b);
}
Complex operator * (const Complex &c) const {
return Complex(a * c.a - b * c.b , a * c.b + b * c.a);
}
};
//len = 2 ^ k
inline void change (Complex y[] , int len) {
for (int i = 1 , j = len / 2 ; i < len -1 ; i ++) {
if (i < j) swap(y[i] , y[j]);
int k = len / 2;
while (j >= k) {
j -= k;
k /= 2;
}
if(j < k) j += k;
}
}
// FFT
// len = 2 ^ k
// on = 1  DFT    on = -1 IDFT
inline void FFT (Complex y[], int len , int on) {
change (y , len);
for (int h = 2 ; h <= len ; h <<= 1) {
Complex wn(cos (-on * 2 * pi / h), sin (-on * 2 * pi / h));
for (int j = 0 ; j < len ; j += h) {
Complex w(1 , 0);
for (int k = j ; k < j + h / 2 ; k ++) {
Complex u = y[k];
Complex t = w * y [k + h / 2];
y[k] = u + t;
y[k + h / 2] = u - t;
w = w * wn;
}
}
}
if (on == -1) {
for (int i = 0 ; i < len ; i ++) {
y[i].a /= len;
}
}
}
int n , m , a
, b
;
int c
, d
, next
, p
;
char str[10];
void get_next (int *a , int l) {
int i = 0 , j = -1;
next[i] = j;
while (i < l) {
if (j == -1 || a[i] == a[j]) {
i ++; j ++;
next[i] = j;
}
else j = next[j];
}
}
int pos
, cnt;
void match (int *a , int la , int *b , int lb) {
int i = 0 , j = 0;
while (i < la) {
if (j == -1 || a[i] == b[j]) {
i ++;j ++;
if (j == lb) {
pos[cnt ++] = i - lb;
j = next[j];
}
}
else j = next[j];
}
}
Complex x1
, x2
;
void gao (int *a , int *b) {
int len = max (n , m);
int l = 1;
while (l < len * 2) l <<= 1;
for (int i = 0 ; i < n ; i ++)
x1[i] = Complex (a[i] , 0);
for (int i = n ; i < l ; i ++)
x1[i] = Complex (0 , 0);
FFT (x1 , l , 1);
for (int i = 0 ; i < m ; i ++)
x2[i] = Complex (b[i] , 0);
for (int i = m ; i < l ; i ++)
x2[i] = Complex (0 , 0);
FFT (x2 , l , 1);
for (int i = 0 ; i < l ; i ++)
x1[i] = x1[i] * x2[i];
FFT (x1 , l , -1);
for (int i = 0 ; i <= n - m ; i ++) {
p[i] += (int)(x1[i + m - 1].a + 0.5);
}
}
int main () {
while (scanf ("%d %d" , &n , &m) != EOF) {
for (int i = 0 ; i < n ; i ++) {
scanf ("%s" , str);
a[i] = 0;
for (int j = 0 ; j < 7 ; j ++)
a[i] = a[i] * 2 + str[j] - '0';
c[i] = str[7] - '0';
}
for (int i = 0 ; i < m ; i ++) {
scanf ("%s" , str);
b[i] = 0;
for (int j = 0 ; j < 7 ; j ++)
b[i] = b[i] * 2 + str[j] - '0';
d[m - i - 1] = str[7] - '0';
}
get_next (b , m);
cnt = 0;
match (a , n , b , m);
if (cnt == 0) puts ("No");
else {
memset (p , 0 , sizeof(p));
puts ("Yes");
gao (c , d);
for (int i = 0 ; i < n ; i ++)
c[i] = c[i] ^ 1;
for (int i = 0 ; i < m ; i ++)
d[i] = d[i] ^ 1;
gao (c , d);
int ans = m + 1 , idx = -1;
for (int i = 0 ; i < cnt ; i ++) {
if (m - p[pos[i]] < ans)
ans = m - p[pos[i]] , idx = pos[i];
}
printf ("%d %d\n" , ans , idx + 1);
}
}
return 0;
}