标签: DP
题目链接
分析
虽然是不难的dp,但是放在F题的位置以为很难,我还是想复杂了结果搞了好久才搞出来,而且还遇到了std::unordered_map
的坑。CF上有组数据专门卡unordered_map
然后就tle了。 因为unordered_map
是hash存储,所以会将hash值相同的放在一个桶里,但是在桶中查找的时候复杂度是$O(n)$的,在codeforces上找到了如下的优化方法
unordered_map<int,int>mp;
mp.reserve(1024);
mp.max_load_factor(0.25);
这样就不会tle了。 具体原理看下面两个链接吧: http://codeforces.com/blog/entry/21853 http://en.cppreference.com/w/cpp/container/unordered_map/reserve
代码
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <map>
#include <unordered_map>
using namespace std;
const int maxn=200050;
int a[maxn];
unordered_map<int,int> dp;
int main(){
dp.reserve(maxn);
dp.max_load_factor(0.25);
int n;
scanf("%d", &n);
int M=0,e=0;
for(int i = 0; i < n; ++i){
scanf("%d", a+i);
int& u=dp[a[i]];
if(dp.count(a[i]-1)) u=max(u,dp[a[i]-1]+1);
else u=1;
if(u>M){
M=u;
e=a[i];
}
}
int b=e-M+1;
cout << M << endl;
for(int i = 0; i < n; ++i){
if(a[i]==b){
printf("%d ", i+1);
b++;
}
}
return 0;
}