1150C

math\构造\贪心

题意:一串数字只含1,2,求前缀和中质素最多的排列方式。
思路:质数分布在2和大于2的奇数中,标程先特判全1和全2
在其他情况下 先放2,1 再全放2 再全放1
我的解法 先放2 1 后再放偶数个1再放全部的2 再放剩下的1 (跟标程原理一样,但标程的更完美,标程取了0为那个偶数个1)
这样可以覆盖到最大值以内所有的素数。

#include<bits/stdc++.h>
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define endl '\n'
#define fi for(int i=1;i<=n;i++) 
#define fi2 for(int i=0;i<n;i++) 
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef set<int> si;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int INF=0x3f3f3f3f; 
const int MAX=2e5+10;
int gcd(int a,int b){
    return b?gcd(b,a%b):a;
}
int a[MAX],sum[3];
vi v;
int main(){
	int n;
	cin>>n;
	
	fi{
		cin>>a[i];
		sum[a[i]]++;
	}
	if(sum[1]==0||sum[2]==0){
		fi{
			cout<<a[i]<<' ';
		}
		return 0;
	}
	v.push_back(2);
	v.push_back(1);
	sum[1]--;
	sum[2]--;
	if(sum[1]%2==0){
	
		for(int i=1;i<=sum[1];i++){
			v.push_back(1);
		}
//21112	
	for(int i=1;i<=sum[2];i++){
		v.push_back(2);
	}}
	else{
		for(int i=1;i<=sum[1]-1;i++){
			v.push_back(1);
		}

	for(int i=1;i<=sum[2];i++){
		v.push_back(2);
	}
	v.push_back(1);
}
	
	for(auto it=v.begin();it!=v.end();it++){
		cout<<*it<<' ';
	}
	
}
#include<bits/stdc++.h>
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define endl '\n'
#define fi for(int i=1;i<=n;i++) 
#define fi2 for(int i=0;i<n;i++) 
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef set<int> si;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int INF=0x3f3f3f3f; 
const int MAX=2e5+10;
int gcd(int a,int b){
    return b?gcd(b,a%b):a;
}

int ans[MAX];
int sum[MAX];

int main() {
//	freopen("in.txt","r",stdin);
	IO;
    int n,x=0,y=0;
    cin>>n; 
    fi2{
        int x;
        cin>>x;
        sum[x]++;
    }
    if(sum[2]){
        y++;
        sum[2]--;
		ans[x++]=2;
        if(sum[1]){
            y++;
			ans[x++]=1;
            sum[1]--;    
        }
    }
    while(1){
        if(sum[2]==0)break;
		ans[x++]=2;
        sum[2]--;
    }
    while(1){
        if(sum[1]==0)break;
		ans[x++]=1;
        sum[1]--;
    }
    for(int i=0;i<x;i++){
		cout<<ans[i]<<' ';
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_43845768/article/details/89705612