Soluție HackerRank pentru Array Construction. Include cerința formatată, exemple, explicația pașilor și cod sursă.
- Problemă: Array Construction
Cerinta completa
Professor GukiZ has hobby — constructing different arrays. His best student, Nenad, gave him the following task that he just can’t manage to solve:
Construct an [Expresie matematică indisponibilă în copia arhivată]-element array, [Expresie matematică indisponibilă în copia arhivată], where the sum of all elements is equal to [Expresie matematică indisponibilă în copia arhivată] and the sum of absolute differences between each pair of elements is equal to [Expresie matematică indisponibilă în copia arhivată]. All elements in [Expresie matematică indisponibilă în copia arhivată] must be non-negative integers.
If there is more then one such array, you need to find the lexicographically smallest one. In the case no such array [Expresie matematică indisponibilă în copia arhivată] exists, print [Expresie matematică indisponibilă în copia arhivată].
Note: An array, [Expresie matematică indisponibilă în copia arhivată], is considered to be lexicographically smaller than another array, [Expresie matematică indisponibilă în copia arhivată], if there is an index [Expresie matematică indisponibilă în copia arhivată] such that [Expresie matematică indisponibilă în copia arhivată] and, for any index [Expresie matematică indisponibilă în copia arhivată], [Expresie matematică indisponibilă în copia arhivată].
Input Format
The first line contains an integer, [Expresie matematică indisponibilă în copia arhivată], denoting the number of queries.
Each of the [Expresie matematică indisponibilă în copia arhivată] subsequent lines contains three space-separated integers describing the respective values of [Expresie matematică indisponibilă în copia arhivată] (the number of elements in array [Expresie matematică indisponibilă în copia arhivată]), [Expresie matematică indisponibilă în copia arhivată] (the sum of elements in [Expresie matematică indisponibilă în copia arhivată]), and [Expresie matematică indisponibilă în copia arhivată] (the sum of absolute differences between each pair of elements).
Constraints
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
Subtasks
For [Expresie matematică indisponibilă în copia arhivată] of the maximum score:
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
For [Expresie matematică indisponibilă în copia arhivată] of the maximum score:
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
- [Expresie matematică indisponibilă în copia arhivată]
Output Format
For each query, print [Expresie matematică indisponibilă în copia arhivată] space-separated integers describing the respective elements of the lexicographically smallest array [Expresie matematică indisponibilă în copia arhivată] satisfying the conditions given above. If no such array exists, print [Expresie matematică indisponibilă în copia arhivată] instead.
Sample Input
1
3 3 4
Sample Output
0 1 2
Explanation
We have [Expresie matematică indisponibilă în copia arhivată] query in which [Expresie matematică indisponibilă în copia arhivată], [Expresie matematică indisponibilă în copia arhivată], and [Expresie matematică indisponibilă în copia arhivată]. The lexicographically smallest array is [Expresie matematică indisponibilă în copia arhivată].
- The sum of array [Expresie matematică indisponibilă în copia arhivată]‘s elements is [Expresie matematică indisponibilă în copia arhivată]
-
The absolute differences between each pair of elements are:
[Expresie matematică indisponibilă în copia arhivată]
[Expresie matematică indisponibilă în copia arhivată]
[Expresie matematică indisponibilă în copia arhivată]The sum of these absolute differences is [Expresie matematică indisponibilă în copia arhivată]
As array [Expresie matematică indisponibilă în copia arhivată] is both lexicographically smallest and satisfies the given conditions, we print its contents on a new line as 0 1 2.
Limbajul de programare folosit: cpp14
Cod:
#include <string>
#include <vector>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<stack>
#include<queue>
#include<cmath>
#include<algorithm>
#include<functional>
#include<list>
#include<deque>
#include<bitset>
#include<set>
#include<map>
#include<unordered_map>
#include<cstring>
#include<sstream>
#include<complex>
#include<iomanip>
#include<numeric>
#include<cassert>
#define X first
#define Y second
#define pb push_back
#define rep(X,Y) for (int (X) = 0;(X) < (Y);++(X))
#define reps(X,S,Y) for (int (X) = S;(X) < (Y);++(X))
#define rrep(X,Y) for (int (X) = (Y)-1;(X) >=0;--(X))
#define repe(X,Y) for ((X) = 0;(X) < (Y);++(X))
#define peat(X,Y) for (;(X) < (Y);++(X))
#define all(X) (X).begin(),(X).end()
#define rall(X) (X).rbegin(),(X).rend()
#define eb emplace_back
#define UNIQUE(X) (X).erase(unique(all(X)),(X).end())
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
template<class T> using vv=vector<vector<T>>;
template<class T> ostream& operator<<(ostream &os, const vector<T> &t) {
os<<"{"; rep(i,t.size()) {os<<t[i]<<",";} os<<"}"<<endl; return os;}
template<class S, class T> ostream& operator<<(ostream &os, const pair<S,T> &t) { return os<<"("<<t.first<<","<<t.second<<")";}
template<class T> inline bool MX(T &l,const T &r){return l<r?l=r,1:0;}
template<class T> inline bool MN(T &l,const T &r){return l>r?l=r,1:0;}
const ll MOD=1e9+7;
int dp[55][222][2222];
int main(){
ios_base::sync_with_stdio(false);
cout<<fixed<<setprecision(0);
fill(dp[0][0],dp[0][0]+55*222*2222,MOD);
dp[0][0][0]=dp[1][0][0]=0;
reps(i,1,51)rep(j,202)rep(k,2004)reps(x,dp[i][j][k],202-j){
const int tmp=k+x*i-j;
assert(tmp>=k);
if(j+x<=202 && tmp<=2002) MN(dp[i+1][j+x][tmp],x);
}
//cout<<"ready"<<endl;
int T;
cin>>T;
while(T--){
int n,s,t;
cin>>n>>s>>t;
//rep(i,n+1){rep(j,s+1){rep(k,t+1)cout<<(dp[i][j][k]==MOD?-1:dp[i][j][k])<<",";cout<<endl;}cout<<endl;}
int o=0;
int ss=s,tt=t;
for(;ss>=0;ss-=n,++o) if(dp[n][ss][tt]<=s) break;
if(ss<0){assert(dp[n][s][t]==MOD);
cout<<-1<<endl;
continue;
}
vector<int> re(n);
rep(i,n){
rep(j,ss+1){
int ns=ss-j*(n-i);
int nt=tt-(ss-j*(n-i));
//cout<<i<<pii(ns,nt)<<endl;
if(nt>=0){
for(int ad=0;ns>=0;ns-=n-i-1,++ad)
if(dp[n-1-i][ns][nt]<MOD){
re[i]=j+o;
o+=ad;
ss=ns; tt=nt;
//cout<<j<<pii(ss,tt)<<endl;
j=MOD; break;
}
}
assert(j!=ss);
}
}
rep(i,n) cout<<re[i]<<" ";cout<<endl;
assert(accumulate(all(re),0)==s);
int sum=0;
rep(i,n)rep(j,i) sum+=re[i]-re[j];
assert(sum==t);
rep(i,n-1) assert(re[i]<=re[i+1]);
assert(re.size()==n);
}
return 0;
}
Scor obtinut: 1.0
Submission ID: 464650223
Link challenge: https://www.hackerrank.com/challenges/array-construction/problem