Codeforces Round #215_div2_D. Sereja ans Anagrams

转载注明出处 http://blog.csdn.net/moedane

 

 

 

传送门 http://codeforces.com/contest/368/problem/D

 

 

题意

给出a,b数组，以及整数p，求所有的q，使得a[q],a[q+p],a[q+2p]...a[q+(m-1)p]等于任意重新排列后的b数组。即b数组中的每一个数都能与上述选出来的数列对应。

 

思路

枚举q从1到p，把a数组中以q为开头的间隔p的所有数取出来作为一个新数列。

初始化时用map<int,int>，把b数组中的数存入map中，然后再在新数列里扫过去，每扫到一个数，就map[a[j]] --，如果长度超过了m，则把a[j - m*p]那个数放回map里（保证map里存的数的数量是m个），这样做的意思是，这时的起点已经由j - m*p 改变到(j - m*p + p）。

每扫一个数还要做的操作是，如果map中某个key的值为0，则erase这个key。这样判断当前起点是否合法就便利了，只要判断当前map是否为空即可（为空则代表选取的这一段恰好能跟b数组一一对应）。

 

T_T这个题T了好久，原因是每一次都做了初始化。后来做一些处理把初始化放到外面，只做一次，就A了。

复杂度似乎是O(nlogn)的，虽然常数大了点……

 

代码

#include <iostream>#include <string>#include <cstring>#include <cstdio>#include <cmath>#include <algorithm>#include <vector>#include <queue>#include <stack>#include <map>#include <cctype>#define bug puts("here");using namespace std;typedef long long ll;const int maxn = 2 * 100086;const int mod = 1000000007;const double PI = atan(1.0)*4.0;map<int,int> mp;int a[maxn],b[maxn];int n,m,p;vector<int> ans;void init(){    mp.clear();    int i;    for(i=1;i<=m;i++)        mp[b[i]] ++;    return;}int main(){    ans.clear();    scanf("%d%d%d",&n,&m,&p);    int i,j;    for(i=1;i<=n;i++)        scanf("%d",a+i);    for(i=1;i<=m;i++)        scanf("%d",b+i);    init();    for(i=1;i<=p;i++)    {        if(i + (m-1) * p > n) break;        int e = 1;        for(j=i;j<=n;j+=p,e++)        {            int tmp = j - m * p;            if(e <= m)            {                mp[a[j]] --;                if(mp[a[j]] == 0) mp.erase(a[j]);            }            else            {                mp[a[tmp]] ++;                mp[a[j]] --;                if(mp[a[j]] == 0) mp.erase(a[j]);                if(mp[a[tmp]] == 0) mp.erase(a[tmp]);            }            if(e>=m && mp.size() == 0) ans.push_back(j - (m-1) * p);            if(j+p > n)            {                for(int k = tmp+p;k<=n;k+=p)                {                    mp[a[k]] ++;                    if(mp[a[k]] == 0) mp.erase(a[k]);                }            }        }    }    sort(ans.begin(),ans.end());    int cnt = ans.size();    printf("%d\n",cnt);    for(i=0;i<cnt - 1;i++)        printf("%d ",ans[i]);    if(cnt > 0) printf("%d\n",ans[i]);}