蒟蒻求优化代码
查看原帖
蒟蒻求优化代码
195331
Mine_KingCattleya楼主2021/6/26 21:43

RT,#1 TLE+MLE,求大佬优化一下这个代码kel或者直接否定掉这个算法
思路是用小根堆存结果,每次取出当前最小的数,然后让它与 p[]p[] 相乘获得新的数再塞回小根堆直到取出第 kk 个数。

#include<map>
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,k,p[105];
int tot;
priority_queue<long long>q;
map<long long,bool>mp;
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&p[i]),q.push(-p[i]);
	while(!q.empty())
	{
		long long now=-q.top();
		q.pop();
		if(mp[now]) continue;
		mp[now]=true;
		if(++tot==k)
		{
			printf("%lld\n",now);
			break;
		}
		for(int i=1;i<=n;i++) q.push(-now*p[i]);
	}
	return 0;
}
2021/6/26 21:43
加载中...