题目链接:https://ac.nowcoder.com/acm/contest/5671/K
题目大意:
一个由一些1-n的排列组成的数列称为k-bag。比如1,2,3,2,1,3,3,2,1就是一个k-bag。部分k-bag是k-bag的一个子串,现在给一个长度为n的数列,判断是否是一个部分k-bag。
解题思路:
参考sunsetcolors的解法:https://blog.nowcoder.net/n/1cf39334dfea43339767441bc1e836b6
跟据题意能想到,满足条件的序列,我们不管头和尾,中间一定要是几个k排列摆放的过程。
但是我们并不清楚,头和尾在什么位置结束。
因此我们考虑一种尺取+dp的策略,首先在序列开头只要不出现重复的元素,那么很明显这些位置都可以作为头部结束的位置,我们将这些位置的dp[i]都设为true,然后我们通过尺取法,快速找到序列里所有满足k排列的哪些区间的结束位置,因为如果我们头部从一个位置pos结束,要是序列符合条件的话必须满足pos+k+k+k...这样的形式,这里面的每个k都是满足条件的k排列区间,假设r是现在一个排列区间的结尾,存在这样的一个转移dp[r]=dp[r-k].
这样实际上我们就是处理出了从头部起,所有符合条件的区间能到达的位置,然后我们再从尾部出发,如果能找到一个尾部开始的位置,和这些从头部起满足条件的位置连接起来,则说明这个数列是一个k-bag的子串。
代码:
#include<bits/stdc++.h> #pragma GCC optimize(3) using namespace std; #define IO std::ios::sync_with_stdio(false); cin.tie(0) const int maxn = 5e5+7; int a[maxn]; bool dp[maxn]; unordered_map<int,int> memo; unordered_set<int> st; int n, k; signed main() { IO; int T; cin >> T; while(T--) { cin >> n >> k; memo.clear(); st.clear(); for(int i = 1; i <= n; i++) cin >> a[i], dp[i] = 0; int pos; for(pos = 1; pos <= n; pos++){ if(memo[a[pos]]) break; memo[a[pos]] = 1; st.insert(a[pos]); } for(int i = 1; i < pos; i++) dp[i] = true; for(int i = pos; i <= min(n,k); i++) { memo[a[i]]++; st.insert(a[i]); } for(int r = k + 1; r <= n; r++) { int l = r - k; //在[l,r]这段区间尺取 memo[a[l]]--; if(!memo[a[l]]) st.erase(a[l]); memo[a[r]]++; st.insert(a[r]); if(st.size() == k) dp[r] |= dp[l]; } memo.clear(); bool v = true; for(pos = n; pos >= 1; pos--) { if(dp[pos]) break; //相接了,说明 if(memo[a[pos]]) { v = false; break; } memo[a[pos]] = 1; } if(v) cout << "YES" <<endl; else cout << "NO" << endl; } return 0; }