Codeforces Round #556 (Div. 2) C - Prefix Sum Primes（思维+贪心）

链接:http://codeforces.com/problemset/problem/1150/C
来源:Codeforces

  给你一个数组，数组中含有只含有1和2，题目要求我们给出一种排列方法使得这个数组的前缀和中含有的素数最多。解法就是我们从素数中可以看出来两个素数之前的差一定是偶数（除了2和3以外），那么此时我们就可以统计1和2的个数，先输出一个2和一个1，然后从剩下的数中先把2输出，在把1输出。

#include<bits/stdc++.h>
using namespace std;

const int inf=0x3f3f3f3f;
const int Max_n=2e5+10;

int a[Max_n];
int n;

int main(){
    scanf("%d",&n);
    int cnt1=0,cnt2=0;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        if(a[i]==1) cnt1++;
        else cnt2++;
    }
    if(cnt2) printf("2"),cnt2--;
    if(cnt1) printf(" 1"),cnt1--;
    while(cnt2) printf(" 2"),cnt2--;
    while(cnt1) printf(" 1"),cnt1--;
    return 0;
}