USTC 1281 Unhappy dots STL map-优快云博客

本文链接：https://blog.youkuaiyun.com/waitfor_/article/details/7214390

本文分享了一道算法竞赛题目中使用map替代multiset解决TLE问题的经验，详细对比了两者在查找和计数操作上的效率差异，并附上了具体的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这题被set坑了，问了问戴牛原来multiset的count 复杂度很大，是logn+o(元素个数)，所以我悲剧的一直TLE啊（除夕夜就这么悲剧....）

其实是用map

#include<stdio.h>
#include<utility>
#include<map>
#include<string.h>
using namespace std;
int x[100005],y[100005];
int visit[100005];
int h[5]={1,-1,0,0,0};
int g[5]={0,0,1,-1,0};
int main(){
    int k,n,m,i,j,u,v;
    int ca=0;
    while(scanf("%d %d %d",&k,&n,&m) && !(k==0 && n==0 && m==0)){
		memset(visit,0,sizeof(visit));
        map<pair<int,int>,int>st;
		for(i=1;i<=k;i++){
			scanf("%d %d",&x[i],&y[i]);
			map<pair<int,int>, int>::iterator it;
			it=st.find(make_pair(x[i],y[i]));
			if(it==st.end())
				st.insert(make_pair(make_pair(x[i],y[i]),1));
			else
				it->second++;
		}
		int sum=0;
		for(i=1;i<=k;i++){
			int tem=0,temp;
			for(int j=0;j<=4;j++){
				int p=x[i]+h[j];
				int q=y[i]+g[j];
				if(p>m || p<1 || q>n ||q<1)
					continue;
				map<pair<int,int>, int>::iterator it;
				it=st.find(make_pair(p,q));
				if(it==st.end())
					continue;
				if(j!=4)
					tem+=it->second;
				else
					tem+=it->second-1;
				if(tem>=2)
					break;
			}
        if(tem<2)
            sum++;
    }
    printf("Case %d: %d\n",++ca,sum);
    }
    return 0;
}

顺便也把我TLE的set贴一下

#include<stdio.h>
#include<utility>
#include<set>
using namespace std;
int x[100005],y[100005];
int h[5]={1,-1,0,0,0};
int g[5]={0,0,1,-1,0};
int main(){
    int k,n,m,i,j,u,v;
    int ca=0;
    while(scanf("%d %d %d",&k,&n,&m) && !(k==0 && n==0 && m==0)){
        multiset<pair<int,int> >st;
    for(i=1;i<=k;i++){
        scanf("%d %d",&x[i],&y[i]);
        st.insert(make_pair(x[i],y[i]));
    }
    int sum=0;
    for(i=1;i<=k;i++){
        int tem=0;
        for(int j=0;j<=4;j++){
            int p=x[i]+h[j];
            int q=y[i]+g[j];
            if(p>m || p<1 || q>n ||q<1)
                continue;
            if(j!=4)
                tem+=st.count(make_pair(p,q));
            else
                tem+=st.count(make_pair(p,q))-1;
            if(tem>=2)
                break;
        }
        if(tem<2)
            sum++;
    }
    printf("Case %d: %d\n",++ca,sum);
    }
    return 0;
}