CFdiv1+2-Restore Permutation-(二分+树状数组维护前缀和)

D

题意：
就是一个数组是全排列，但是只给你说这个点前面所有比他小的数的总和室多少。现在让你根据条件把这个全排列求出来。

思考：
刚开始看到就感觉要么是从前往后走，要么是从后往前。我推了推，还是从后往前走好算，从后往前填。比如填到i，set里面还剩下一堆数，这个va[i]肯定是这个set里面的前缀和，所以找到这个前缀和后面的那个数就行了。但是如果用set每次这样操作，是nn的复杂度。然后我想了想，既然是求前缀和，可以用个树状数组来维护呀，这样每次删的时候也直接删就行了。找前缀和=va[i]的可以直接二分找，因为前缀和肯定是单调递增的。所以这题就结束了，复杂度<=nlog(n)*log(n)。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
		
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
int va[N];
int anw[N];

int tr[N],R = 2e5+5;

int bit(int x)
{
	return x&(-x);
}

void update(int x,int value)
{
	while(x<=R)
	{
		tr[x] += value;
		x += bit(x);
	}	
}

int query(int x)
{
	int sum = 0;
	while(x)
	{
		sum += tr[x];
		x -= bit(x);
	}
	return sum;
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<=n;i++) update(i,i);
	for(int i=n;i>=1;i--)
	{
		int l = 1,r = n;
		while(l<r)
		{
			int mid = (l+r+1)>>1;
			if(query(mid-1)<=va[i]) l = mid;
			else r = mid-1;
		}
		anw[i] = l;
		update(l,-l);
	}
	for(int i=1;i<=n;i++) cout<<anw[i]<<" ";
	return 0;
}

总结：
对于题目一点一点去分析，是解决的关键。