题意:
你有一个初始分数,有k个评委按顺序依次对你的分数进行修改,结束后你记得n个你的分数修改过程中出现过的值。
询问你的初始分数可能有多少种情况。
(n,k<=2000)
思路
因为你记得的分数一定出现在1~k次修改后的某一个环节
一旦确定某个环节后的分数,就能o(k)的确定所有环节出现的分数
那么枚举记的第一个数出现的位置(1~k),然后算出在这种情况下每个位置的数,用map记录出现的数出现的下标(如果没有出现就为0),这样就可以O(n*log(n))的判断这种情况是否合法(是否记得的每个数都出现过)。
总的复杂度为O(k*k*log(n))属于O(能过)。
但是交上去一直T,后来发现是在使用map的时候习惯用 [ ] 去查询对应的值
但是使用 [ ] 查询的原理是这样的:
如果查询的数x是map的key_value,那么就是正常的操作
但是如果查询过的数x不是key_value,那么它会插入一个(x,0)的新元素
因为这道题需要用到大量的实值为0的查询,频繁使用 [ ] 会最坏的情况下可能插入k*k个新元素,于是出现了超时的情况。
解决方法也很简单,就是使用map的find函数进行查询
find函数的返回值是查询map的迭代器,在查询时不会插入新的元素
如果容器中不存在查询的key_value,那么就会返回end()迭代器。
这样写出来就是正常的O(k*k*log(n))该有的时间(156ms)了。
代码:
#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
int n,k;
int A[2005];
int B[2005];
int fro[2005];
map<int,int>id;
set<int>S;
bool used[2005];
void work(int pos){
memset(used,0,sizeof(used));
used[1] = true;
int res = 1;
int a = B[1] , b = B[1];
for(int i=pos+1;i<=n&&res<k;i++){
b += A[i];
auto it = id.find(b);
if(it!=id.end()&&!used[it->se]){
used[it->se] = true;
res++;
}
}for(int i=pos;i>1&&res<k;i--){
a -= A[i];
auto it = id.find(a);
if(it!=id.end()&&!used[it->se]){
used[it->se] = true;
res++;
}
}if(res==k)S.insert(B[1]-fro[pos]);
}
inline int read()
{
int x=0,f=1; char ch=getchar();
while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
int main()
{
n = read(); k = read();
for(int i=1;i<=n;i++){
A[i] = read();
fro[i] = fro[i-1] + A[i];
}for(int i=1;i<=k;i++){
B[i] = read();
id[B[i]] = i;
}for(int i=1;i<=n;i++){
work(i);
}printf("%d\n",(int)S.size());
return 0;
}