cf 367B Sereja ans Anagrams

    xiaoxiao2021-03-25  48

    Sereja has two sequences a and b and number p. Sequence a consists of n integers a1, a2, …, an. Similarly, sequence b consists of m integers b1, b2, …, bm. As usual, Sereja studies the sequences he has. Today he wants to find the number of positions q (q + (m - 1)·p ≤ n; q ≥ 1), such that sequence b can be obtained from sequence aq, aq + p, aq + 2p, …, aq + (m - 1)p by rearranging elements.

    Sereja needs to rush to the gym, so he asked to find all the described positions of q.

    Input The first line contains three integers n, m and p (1 ≤ n, m ≤ 2·105, 1 ≤ p ≤ 2·105). The next line contains n integers a1, a2, …, an (1 ≤ ai ≤ 109). The next line contains m integers b1, b2, …, bm (1 ≤ bi ≤ 109).

    Output In the first line print the number of valid qs. In the second line, print the valid values in the increasing order.

    Example Input 5 3 1 1 2 3 2 1 1 2 3 Output 2 1 3 Input 6 3 2 1 3 2 2 3 1 1 2 3 Output 2 1 2

    题意 给出a b两个字符串,还有k间隔。 从a中把每k个间隔的字符连在一起(随便排序),问能组成b的起点位置有多少个。stl的应用,关于map,当map里面的元素相等而且个数相等的时候map类型会完全相等。当元素为0的时候一定要用st.erase(元素值)删除。这样时间复杂度是on

    #include <bits/stdc++.h> using namespace std; #define fi first #define se second #define ff(i,a,b) for(int i = a; i <= b; i++) #define f(i,a,b) for(int i = a; i < b; i++) typedef pair<int,int> P; #define ll long long int a[200010]; map<int,int> mp,tp; int n,m,p; vector<int> ans; queue<int> Q; void solve(int st) { tp.clear(); while(!Q.empty()) Q.pop(); for(int i = st; i <= n; i += p) { tp[a[i]]++; Q.push(i); if(Q.size() == m) { if(tp == mp) ans.push_back(Q.front()); tp[a[Q.front()]]--; if(tp[a[Q.front()]] == 0) tp.erase(a[Q.front()]); Q.pop(); } } } int main() { ios::sync_with_stdio(false); cin >> n >> m >> p; ff(i,1,n) cin >> a[i]; ff(i,1,m) { int x; cin >> x; mp[x]++; } ff(i,1,p) solve(i); sort(ans.begin(),ans.end()); // ans.erase(unique(ans.begin(),ans.end()),ans.end()); cout << ans.size() << endl; if(ans.size()) { printf("%d",ans[0] ); f(i,1,ans.size()) printf(" %d",ans[i]); puts(""); } return 0; }
    转载请注明原文地址: https://ju.6miu.com/read-26058.html

    最新回复(0)