1701D - Permutation Restoration

E. Text Editor

分析：

每个 $i$ 都有 $b_i=\lfloor \frac{i}{a_i}\rfloor$ ，我们可以将其改写如下： $a_i\cdot b_i\le i<a_i\cdot (b_i+1)$, or $\frac{i}{b_i+1}<a_i\le \frac{i}{b_i}$。从这里我们可以看出，每一个 $i$ 都有一段数值可以分配给 $a_i$ 。因此，我们必须将 $1$ 至 $n$ 中的每个数字与其中的一个值段匹配起来。

因此，我们对左端点排序，然后贪心地考虑，每次选择右端点最小的线段，这样的话其他线段有更多的选择空间。从$1$ 枚举到 $n$，当枚举到 $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]<<" \n"[i==n];
   return ;
  }
int main()
{
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>T;
    while(T--) 
    solve();
    return 0;
}