LightOJ 1042 Secret Origins 贪心

题目：

http://www.lightoj.com/volume_showproblem.php?problem=1042

题意：

给出一个数n，求化成二进制后1的个数和n一样多的且比n大的最小的那个数，直接输出

思路：

写了一个巨搓的贪心。。。首先把n分解为二进制数，然后分为两种情况：
1. 所有的1是连续的
此时应该在第一个1后面插入一个0，为了最小，应当把此位置之后的1尽量往后面放
2. 所有的1不连续
此时应该找到第一个这样的情况：…01…，交换01，即变成了…10…，为了最小，把这两个位置之后的1也尽量往后面放

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

typedef long long ll;
const int N = 110;
int cas = 0;
int arr[N];
int main()
{
    int t, n;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        int k = 0, m = n;
        while(m != 0) arr[k++] = m % 2, m /= 2;
        reverse(arr, arr + k);
        int num0 = 0, num1 = 0, f = 0;
        for(int i = k-1; i >= 0; i--)
        {
            if(arr[i] == 1 && !f) f = i;
            if(arr[i]) num1++;
            else num0++;
        }
        if(f + 1 == num1)//所有的1连续
        {
            for(int i = k-1; i >= 1; i--) arr[i+1] = arr[i];
            arr[1] = 0;
            k++;
            int en = k-1;
            for(int i = 2; i < k && i < en; i++)//把下标1之后的1尽量往后放
            {
                while(arr[en] == 1) en--;
                if(i >= en) break;
                if(arr[i] == 1) swap(arr[i], arr[en]);
            }
        }
        else //不连续
        {
            int idx;
            for(int i = k-2; i >= 0; i--)
                if(arr[i] == 0 && arr[i+1] == 1)
                {
                    idx = i;
                    swap(arr[i], arr[i+1]);
                    break;
                }
            int en = k-1;
            for(int i = idx + 1; i < k && i < en; i++)//把交换位置之后的1尽量往后放
            {
                while(arr[en] == 1) en--;
                if(i >= en) break;
                if(arr[i] == 1) swap(arr[i], arr[en]);
            }
        }
        ll res = 0;
        for(int i = 0; i < k; i++) res = res * 2 + arr[i];
        printf("Case %d: %lld\n", ++cas, res);
    }
    return 0;
}