题意:给你一个长度为n的序列s,再给你一个长度为m的序列p,满足m<n,若,s[i.....i+m-1]=p,则我们称p为s的子序列且i为p出现的位置,现在,要求我们对p中的元素进行任意排列组合称为newp,并在s中找到newp第一次出现的位置i,计算所有位置的平方和。
题解:首先,由于对p任意排列组合,并在s中找到它,那么,我们可以不用去管p具体是怎么排列组合的,只需要知道p中的元素及其个数就行了,用map[si]记录s[i]所对应的元素出现的次数,则当统计完后,map的size就是p中不同元素的个数。这样我们对s从左往右扫描从头开始扫描。
定义这个扫描框的长度为m,其右端点为头,左端点为尾,同时定义一个cnt,记录p中与s在扫描框内不同元素的个数。
先大题说下流程。
那么当我们扫描的时候。
每一次右移头先扫描一个数St,则若map[St]=1,那么cnt--,若map[St]==0,则cnt++。
然后map[st]--。
接下来判断尾是否扫描了一个数(因为是从下标0开始扫描的,所以在前m个数中只有头没有尾)
若有尾扫描了一个数Sw,即从扫描框中拿出去一个数Sw,那么若map[Sw]==-1,cnt--,若map[Sw]==0,cnt++.接着map[Sw]++.
在每一步的最后,若cnt==0,则说明,在扫描框中的串与s中的一段匹配,则记录下这个串的位置。
下面说明。
对于一个p中的每个元素pi,map[pi]记录了它在p中出现的次数,那么对于一个空的长度为m的扫描框,我们只需要再向其中加入map[pi]个pi(i=1~map.size())元素,就可使扫描框内的元素构成p的一种排列组合。
那么cnt初始化为map.size就意味着我们还要cnt种不同的元素才有可能凑成p的一种排列。
那么对于每一次扫描,头加进来一个数st,就意味着需要的st元素少了一个(这就是为什么每次头判定的最后需要map[st]--),若此时其map值为一,那么说明这个st就是我们需要的最后一个st,即我们需要的不同元素少了一个,即cnt--。
而对于尾也是一样,具体看程序就明白了。
那么若加进来的元素st不是p中的元素怎么办呢。对于不是p中的元素,map的初始值为0,而由上面流程,map为0是,cnt++,即,扫描框中的不同元素多了一个,若之前扫描框为空,则此时cnt=m+1,意味着多了一个不需要的元素。而进行头判定后,map--变为-1,当下一次尾读到该元素的时候,相应的map为-1时,cnt--,意味着把这个不需要的元素去掉,而这个元素在扫描框内时,cnt是多了一的,无法变成0,所以,不会出现误判。
由上述流程我们知道,一个p中元素sp的map值一直在-1到k(k为sp在p中出现的次数)内变化,而一个非p中元素的map值非-1即0,所以它只会影响cnt的上限,而不会影响cnt的下限。
哎。蒟蒻就是蒟蒻,连话都讲不明白,小伙伴们还是看代码吧。
代码:
#include<bits/stdc++.h>
#define MEM(a,x) memset(a,x,sizeof(a));
#define MEMINF(a) memset(a,0x3f,sizeof(a));
using namespace std;
typedef long long LL;
const int MAXN=50000;
const int INF=0x3f3f3f3f;
const int MOD=1000000007;
int s[MAXN],p;
int main() {
int Test;
int n,m;
int cnt;
LL ans;
cin>>Test;
for (int cas=1; cas<=Test; ++cas) {
ans=0;
map<int,int>mp;
cin>>n>>m;
for (int i=0; i<n; ++i) scanf("%d",&s[i]);
for (int i=0; i<m; ++i) scanf("%d",&p),mp[p]++;
cnt=(int)mp.size();
for (int i=0; i<n; ++i) {
if (mp[s[i]]==1) cnt--;
else if (mp[s[i]]==0) cnt++;
mp[s[i]]--;
if (i-m>=0) {
if (mp[s[i-m]]==0) cnt++;
else if (mp[s[i-m]]==-1) cnt--;
mp[s[i-m]]++;
}
if (cnt==0) {
LL pos=i-m+2;
ans+=pos*pos;
}
}
cout<<ans<<endl;
}
}