HDU 6865 - Kidnapper's Matching Problem

題意

給定\(n,m,k(m\leq n)\)，給定長度為\(n\)的陣列\(\{a\}\)，長度為\(m\)的陣列\(\{b\}\)，長度為\(k\)的集合\(\{S\}\)

定義\(2_⊕^S\)為\(\{S\}\)的任意子集的異或和（xor_sum）的可能結果所形成的新集合

定義兩等長集合\(\{x\},\{y\}\)的\(matches\)函式

該函式會遍歷判斷\(x_i\ xor\ y_i\ \in\ 2_⊕^S\)

如果對於所有的\(i(i\leq|x|,|y|)\)，\(x_i\ xor\ y_i\ \in\ 2_⊕^S\)

問下式的值

\[\sum_{i=1}^{n-m+1}[(a_i,a_{i+1},\dots ,a_{i+m-1})\ matches\ b]\cdot2^{i-1}\ mod\ (10^9+7) \]

限制

\(1\leq T\leq 2\cdot10^4\)

\(1\leq n\leq2\cdot10^5,\ 1\leq m\leq \min(n,\ 5\cdot10^4),\ 1\leq k\leq100\)

\(0\leq a_i,b_i,S_i\lt2^{30}\)

\(\sum n\leq1.2\cdot10^6,\ \sum m\leq3\cdot10^5,\ \sum k\leq 6\cdot10^5\)

思路

先求出\(\{S\}\)的線性基\(bs\)

（下面是官方題解的內容）

結論：假設\(x,y\)消去線性基\(bs\)中的位後得到的數分別位\(x',y'\)，那麼下列關係式成立

\[x⊕y\in 2_⊕^S \iff x'=y' \]

證明：

充分性：根據線性基的性質，\(x⊕y\)必然能由\(x'⊕y'\)再異或上線性基\(bs\)內某些數得到，所以在\(x'=y'\)時\(x⊕y\in 2_⊕^S\)

必要性：反證，因為\(x',y'\)不包含線性基\(bs\)中的位（對應的位被消去），所以\(x'⊕y'\)也不會包含\(bs\)中的位。又因為\(x'\neq y'\)，所以\(x'⊕y'\neq0\)

也確實只要有了這個結論，讓\(\{a\}\)與\(\{b\}\)中所有元素全部消去線性基\(bs\)中的位後，本題便變成了一道快速匹配模板題

由於\(n\geq m\)，且答案是根據\([Sub(a)\ matches\ b]\)來計算的，故選擇\(\{a\}\)作為主串，選擇\(\{b\}\)作為模式串，套KMP即可

主要注意對於題目公式中的\(i=1\)，實際上在KMP中主串游標應該是位於\(i=m\)位置才對

所以如果在主串第\(p\)個位置匹配成功，實際上應該加上\(2^{p-m+1}\)

由於按照順序來匹配，\(2^n\)通過遞推即可，其餘的注意下細節就可以了

程式碼

(826ms/2000ms)

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;

int n,m,k,a[200050],b[200050],S[111];
int Next[200050];

struct LinearBase
{
    int v[35];
    void init()
    {
        memset(v,0,sizeof v);
    }
    void insert(int x)
    {
        for(int i=30;i>=0;i--)
            if(x>>i&1)
            {
                if(v[i])
                    x^=v[i];
                else
                {
                    v[i]=x; 
                    break;
                }
            }
    }
    int solve(int x)
    {
        for(int i=30;i>=0;i--)
            if(x>>i&1)
                x^=v[i];
        return x;
    }
}bs;

void getNext()
{
    Next[1]=0;
    for(int i=2,j=0;i<=m;i++)
    {
        while(j>0&&b[j+1]!=b[i])
            j=Next[j];
        if(b[j+1]==b[i])
            j++;
        Next[i]=j;
    }
}

void KMP()
{
    int n2=1,ans=0;
    for(int i=1,j=0;i<=n;i++)
    {
        while(j>0&&b[j+1]!=a[i])
            j=Next[j];
        if(b[j+1]==a[i])
            j++;
        if(j==m)
        {
            ans=(ans+n2)%mod;
            j=Next[j];
        }
        if(i>=m)
            n2=(2LL*n2)%mod;
    }
    printf("%d\n",ans);
}

void solve()
{
    scanf("%d%d%d",&n,&m,&k);
    bs.init();
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)
        scanf("%d",&b[i]);
    for(int i=1;i<=k;i++)
    {
        scanf("%d",&S[i]);
        bs.insert(S[i]);
    }
    for(int i=1;i<=n;i++)
        a[i]=bs.solve(a[i]);
    for(int i=1;i<=m;i++)
        b[i]=bs.solve(b[i]);
    getNext();
    KMP();
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
        solve();
    return 0;
}