关闭

Codeforces Round #215_div2_D. Sereja ans Anagrams

标签: ACMcodeforcesCF解题报告368D
422人阅读 评论(0) 收藏 举报
分类:

转载注明出处  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数组中的每一个数都能与上述选出来的数列对应。

 

思路

枚举q1p,把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]);
}


 

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:7100次
    • 积分:173
    • 等级:
    • 排名:千里之外
    • 原创:11篇
    • 转载:0篇
    • 译文:0篇
    • 评论:0条