浙江省赛C Array in the Pocket(贪心)

题意:给一个数组A,求一个B,使的B是A的全排列,且对任意位置a[i]!=a[j],求满足条件字典序最小的B数组。

思路:我们从头开始,每次放最小的元素,判断所有元素剩余个数加上A数组当前位置后面所有元素剩余个数是否小于n-i即可。用set+优先队列瞎搞就行了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[100005];
int b[100005];
int m[100005];
int m1[100005];
int ans[100005];
set<int> s;
struct node{
	int val,pos;
	node(){
	}
	node(int q,int w)
	{
		val=q;pos=w;
	}
	bool friend operator <(node q,node w)
	{
		return q.val<w.val;
	}
};
int p[100005];
int vis[100005];
int main()
{
	ios::sync_with_stdio(false);
	int t;
	cin>>t;
	while(t--)
	{
		priority_queue<node> q;
		int n;
		cin>>n;memset(m,0,sizeof(m));
		memset(m1,0,sizeof(m1));memset(p,0,sizeof(p));memset(vis,0,sizeof(vis));
		s.clear();
		
		for(int i=1; i<=n; i++)
		{
			cin>>a[i];
			m[a[i]]++;
			m1[a[i]]++;
			s.insert(a[i]);
		}
		for(int i=1; i<=n; i++)
		{
			int val=(a[i]);
			int key=m[a[i]];
			if(vis[val]==0)
			{
				q.push(node(key+m1[val],val));
				vis[val]=1;
			}
		}
		int flag=1;
		for(int i=1;i<=n;i++)
		{
			set<int>::iterator it=s.begin();
			int f=0;
			while(it!=s.end())
			{
				int val=(*it);
				int key=m[(*it)];
				if(val!=a[i]&&key-1+m1[val]<=n-i)
				{
					while(!q.empty())
					{
						node u=q.top();q.pop();
						u.val-=p[u.pos];
						q.push(u);
						if(p[u.pos]==0)
						{
							int l=u.val;
							if(u.pos==a[i]||u.pos==val) l--;
							if(l<=n-i)
							{
								f=1;break;
							}
							else{
								f=0;break;
							}						
						}
						p[u.pos]=0;
					}
					if(f==1)
					{
						ans[i]=val;
						m[val]--;
						p[val]++;
						if(key-1==0)
						{
							s.erase(it);
						}					
					}
				}
				it++;
				if(f==1) break;
			}
			if(f==0)
			{
				flag=0;break;
			}
			m1[a[i]]--;
			p[a[i]]++;
		}
		if(flag==0) printf("Impossible\n");
		else
		{
			for(int i=1;i<=n;i++) printf("%d%c",ans[i],i==n?'\n':' ');
		}
	}
	return 0;
}
发布了155 篇原创文章 · 获赞 32 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_37632935/article/details/89639569