UVA 11525 - Permutation(樹狀數組)

標籤：style   blog   http   color   os   io   for   2014   

UVA 11525 - Permutation

題目連結

題意：給定一個k個數字，求第n個全排列，由於n很大，輸入的方式為∑k1Si?(K?i)!

思路：全排列，很容易看出，前面的si對應的就是數組中第k小的數字，那麼問題變成每次找第k小的數字，然後去掉這個數字，這個用樹狀數組很容易實現

代碼：

#include <cstdio>#include <cstring>#define lowbit(x) (x&(-x))const int N = 50005;int t, k, bit[N];void add(int x, int v) {    while (x <= k) {bit[x] += v;x += lowbit(x);    }}int get(int x) {    int ans = 0, num = 0;    for (int i = 16; i >= 0; i--) {ans += (1<<i);if (ans >= k || bit[ans] + num >= x)    ans -= (1<<i);else num += bit[ans];    }    return ans + 1;}int main() {    scanf("%d", &t);    while (t--) {memset(bit, 0, sizeof(bit));scanf("%d", &k);for (int i = 1; i <= k; i++)    add(i, 1);int num;for (int i = 1; i <= k; i++) {    scanf("%d", &num);    num++;    int v = get(num);    printf("%d%c", v, i == k ? '\n' : ' ');    add(v, -1);}    }    return 0;}