【CodeForces 297C】Splitting the Uniqueness

#include<cstdio>
#include<algorithm>
using namespace std;

const int N=1e5+10;

struct p
{
    int v,id,s;
} a[N];
bool cmp(p a,p b)
{
    return a.v<b.v;
}
bool cmp1(p a,p b)
{
    return a.id<b.id;
}
int n;
int main()
{
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i].v);
        a[i].id=i;
    }
    printf("YES\n");
    sort(a+1,a+1+n,cmp);
    int k=n/3;
    if(n%3) k++;
    for(int i=1; i<=k; i++)
    {
        a[i].s=0;
        a[i+k].s=a[i+k].v;
        a[i+k*2].s=k-i;
    }
    sort(a+1,a+1+n,cmp1);
    for(int i=1; i<n; i++)
    {
        printf("%d ",a[i].s);
    }
    printf("%d\n",a[n].s);
    for(int i=1; i<n; i++)
    {
        printf("%d ",a[i].v-a[i].s);
    }
    printf("%d\n",a[n].v-a[n].s);
    return 0;
}