1701D – Permutation Restoration

E. Text Editor

分析:

每个

i

i

i 都有

b

i

=

?

i

a

i

?

b_i = leftlfloor frac{i}{a_i}
ight
floor

bi?=?ai?i?? ,我们可以将其改写如下:

a

i

?

b

i

i

<

a

i

?

(

b

i

+

1

)

a_i cdot b_i le i < a_i cdot (b_i+1)

ai??bi?≤i<ai??(bi?+1), or

i

b

i

+

1

<

a

i

i

b

i

frac{i}{b_i + 1} < a_i le frac{i}{b_i}

bi?+1i?<ai?≤bi?i?。从这里我们可以看出,每一个

i

i

i 都有一段数值可以分配给

a

i

a_i

ai? 。因此,我们必须将

1

1

1 至

n

n

n 中的每个数字与其中的一个值段匹配起来。

因此,我们对左端点排序,然后贪心地考虑,每次选择右端点最小的线段,这样的话其他线段有更多的选择空间。从

1

1

1 枚举到

n

n

n,当枚举到

i

i

i 时,将每一个以

i

i

i 为左端点的线段,加入到一个以右端点排序的堆中,每次选择最小的就行了。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10,M=1e9+10;
typedef long long ll;
typedef pair<int,int> PII;
int T;
vector<PII> g[N];
priority_queue<PII,vector<PII>,greater<PII>> heap;
int ans[N];
void solve()
{
   int n;
   cin>>n;
   int b[N];
   for(int i=1;i<=n;i++) cin>>b[i],g[i].clear();
   for(int i=1;i<=n;i++)
   {
     int l,r;
     if(b[i]==0) l=i+1,r=n;
     else l=i/(b[i]+1) + 1,r=i/b[i];
     g[l].push_back({r,i});
   }
   for(int i=1;i<=n;i++)
   {
    for(auto x : g[i])
       heap.push(x);
    auto t=heap.top();
    heap.pop();
    ans[t.second]=i; 
   }
   for(int i=1;i<=n;i++) cout<<ans[i]<<" 
"[i==n];
   return ;
  }
int main()
{
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>T;
    while(T--) 
    solve();
    return 0;
}