码迷,mamicode.com
首页 > 其他好文 > 详细

UVA 11525 - Permutation(树状数组)

时间:2014-07-30 17:25:44      阅读:167      评论:0      收藏:0      [点我收藏+]

标签: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;
}


UVA 11525 - Permutation(树状数组),布布扣,bubuko.com

UVA 11525 - Permutation(树状数组)

标签:style   blog   http   color   os   io   for   2014   

原文地址:http://blog.csdn.net/accelerator_/article/details/38303573

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!