解题报告：Codeforces Round #424 Div. 2 C. Jury Marks 记录STL::map的一个小细节

题目链接

题意：

你有一个初始分数，有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;
}