2017年9月23日提高组T1 奶牛编号

Description

---

作为一个神秘的电脑高手，Farmer John 用二进制数字标识他的奶牛。

然而，他有点迷信，标识奶牛用的二进制数字，必须只含有K位“1” (1 <= K <= 10)。 当然，每个标识数字的首位必须为“1”。  

FJ按递增的顺序，安排标识数字，开始是最小可行的标识数字（由“1”组成的一个K位数）。  

不幸的是，他没有记录下标识数字。请帮他计算，第N个标识数字 (1 <= N <= 10^7)。   

Input

---

第1行：空格隔开的两个整数，N和K。

Output

---

第N个标识数字

Source

---

LZH

Solution

---

显然我们只关心1在哪里，那么我们用是个数字记录每个1的位置。每次从右往左找到第一个可以向右移动的1，把右边的1全部移到最右边，这样我们就得到了一个比原来大1的二进制序列。这样我们就在O(nk)的时间内搞定了

Code

---

#include <stdio.h>
#include <string.h>
#include <string>
#include <math.h>
#include <iostream>
#include <algorithm>
#include <stack>
#define rep(i, st, ed) for (int i = st; i <= ed; i += 1)
#define drp(i, st, ed) for (int i = st; i >= ed; i -= 1)
#define erg(i, st) for (int i = ls[st]; i; i = e[i].next)
#define fill(x, t) memset(x, t, sizeof(x))
#define max(x, y) (x)>(y)?(x):(y)
#define min(x, y) (x)<(y)?(x):(y)
#define abs(x) (x)<(0)?(-(x)):(x)
#define INF 0x3f3f3f3f
#define N 1001
#define E 1001
#define L 1001
int n, k;
int pos[20];
inline int read() {
    int x = 0; char ch = getchar();
    for(; ch<'0'||ch>'9'; ch=getchar());
    for(; ch<='9'&&ch>='0'; (x*=10)+=ch-'0',ch=getchar());
    return x;
}
int main(void) {
    n = read();
    k = read();
    rep(i, 1, k) {
        pos[i] = i;
    }
    while (-- n) {
        bool flag = false;
        rep(i, 1, k) {
            if (pos[i] + 1 < pos[i + 1]) {
                pos[i] += 1;
                rep(j, 1, i - 1) {
                    pos[j] = j;
                }
                flag = true;
                break;
            }
        }
        if (!flag) {
            pos[k] += 1;
            rep(i, 1, k - 1) {
                pos[i] = i;
            }
        }
    }
    int cnt = k;
    drp(i, pos[k], 1) {
        if (i == pos[cnt]) {
            cnt -= 1;
            printf("1");
        }else {
            printf("0");
        }
    }
    puts("");
    return 0;
}