【暴力 约束】D - K-th K AtCoder - agc008_d

题面

https://vjudge.net/problem/AtCoder-agc008_d

题意

给一个长度为N的序列X,X[i]表示的是i出现第i次在x[i]这个位置上，问是否能构造出这个序列

思路

首先x[i]这个位置上放的就是i这个数，然后就是约束前面i-1个位置来放i这个数，有一个显然的结论就是贪心地来选肯定是前面的位置先安排掉，因为对于后面的数来说，可以选择的位置更多，同样，在第一轮从前往后排的时候已经占掉前一半的数了，那么后一半的位置就倒着按照第一次的贪心思路来把后面的位置填充完

代码

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int maxn=1e6+10;
const int ha=3e5+10;
int n;
P x[maxn];
int pla[maxn];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
    {
        scanf("%d",&x[i].first);
        x[i].second=i;
    }
	sort(x+1,x+1+n);
	int flag=1;
	for(int i=1;i<=n;i++)
    {
        if(pla[x[i].first]) {flag=0;break;}
        pla[x[i].first]=x[i].second;
    }
    for(int i=1;i<=n;i++)
    {
        int num=x[i].second;
        int k=0;
        for(int j=1;j<x[i].first;j++)
        {
            if(k==num-1) break;
            if(pla[j]==0)
            {
                pla[j]=num;
                k++;
            }
        }
        if(k<num-1)
        {
            flag=0;break;
        }
    }
    for(int i=n;i>=1;i--)
    {
        int num=x[i].second;
        int k=0;
        for(int j=n*n;j>x[i].first;j--)
        {
            if(k==n-num) break;
            if(pla[j]==0)
            {
                pla[j]=num;
                k++;
            }
        }
        if(k<n-num)
        {
            flag=0;break;
        }
    }
    if(!flag) printf("No\n");
    else
    {
        printf("Yes\n");
        for(int i=1;i<=n*n;i++)
        {
            if(i!=1) printf(" ");
            printf("%d",pla[i]);
        }
    }
	return 0;
}