小红的排列构造（dp优化）

最新推荐文章于 2025-04-14 22:38:15 发布

原创最新推荐文章于 2025-04-14 22:38:15 发布 · 1k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #动态规划 #c++

C++ 专栏收录该内容

63 篇文章

订阅专栏

文章讨论了如何针对输入描述中的问题提供一种更高效的解法，主要关注优化代码以减少内存使用和运行时间。

题目描述

小红拿到了一个长度为n的数组a，她希望你构造两个排列p和q，满足对于i∈[1,n],ai∈[1,n]pi或qi二选一。你能帮帮她吗？定义排列是一个长度为n的数组，其中1到n每个元素恰好出现1次。

输入描述:第一行输入一个正整数n，代表两个数组的长度。第二行输入n个正整数ai。1≤n≤10^5,1≤ai≤n。

输出描述:如果无解，请输出-1。否则第一行输出n个正整数pi，第二行输出n个正整数qi，代表小红构造的两个排列。有多解时输出任意即可。

输入

2 3 2

输出

2 3 1
1 3 2

输入

4
1 1 1 1
输出：-1

#include<bits/stdc++.h>
#include<iostream>
#include<vector>
using namespace std;
const long long N=1e7+5;
inline int read(){//内联函数效率更高
    int x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-')
            f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=(x<<1)+(x<<3)+(c^48),c=getchar();
    }
    return x*f;
}
inline void print(int x)
{
    if(x<0)
        putchar('-'),x=-x;
    if(x>9)
        print(x/10);
    putchar(x%10+'0');
    return;}
int n,k=0,z=0;
vector<int> tr(N),arr(N),p1(N),p2(N);//向量比数组更加灵活与方便
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		arr[i]=read();
		if(tr[arr[i]]==2){
			cout<<"-1";
			return 0;
		}
        ++tr[arr[i]];
	}
	for(int i=1;i<=n;i++){
		if(tr[arr[i]]==1){
			p1[i]=arr[i];
            p2[i]=arr[i];
		}
		if(tr[arr[i]]==3){
			while(tr[++k]!=0);
				p1[i]=k;
			p2[i]=arr[i];
		}
		if(tr[arr[i]]==2){
			tr[arr[i]]=3;
			p1[i]=arr[i];
			while(tr[++z]!=0);
				p2[i]=z;
		}
	}	
	for(int i=1;i<=n;i++)print(p1[i]),putchar(32);
	putchar(10);
	for(int i=1;i<=n;i++)print(p2[i]),putchar(32);
	putchar(10);	
    return 0;
}

这个虽然通过运行但是时间和内存都耗费很大

#include<bits/stdc++.h>
#define ll long long
const ll N=1e5+5;
#define _for(i,a,b) for(int(i)=(a);(i)<(b);(i)++)
using namespace std;
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<<1)+(x<<3)+(ch^48),ch=getchar();
    }
    return x*f;
}
void write(int x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
    return;}
int n,k=0,z=0;
vector<int> tr(N),arr(N),p1(N),p2(N);
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		arr[i]=read();
		if(tr[arr[i]]==2){
			cout<<"-1";
			return 0;
		}
        ++tr[arr[i]];
	}
	for(int i=1;i<=n;i++){
		if(tr[arr[i]]==1){
			p1[i]=arr[i];
            p2[i]=arr[i];
		}
		if(tr[arr[i]]==3){
			while(tr[++k]!=0);
				p1[i]=k;
			p2[i]=arr[i];
		}
		if(tr[arr[i]]==2){
			tr[arr[i]]=3;
			p1[i]=arr[i];
			while(tr[++z]!=0);
				p2[i]=z;
		}
	}	
	for(int i=1;i<=n;i++)write(p1[i]),putchar(32);
	putchar(10);
	for(int i=1;i<=n;i++)write(p2[i]),putchar(32);
	putchar(10);		
    return 0;
}