cpp/A_A_characteristic.cpp at main · Sayemahamed/cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include <bits/stdc++.h>
using namespace std;
//----------------------------------------------------------------//
// definitions //
#define endl '\n'
//----------------------------------------------------------------//
// preDefined functions//

//----------------------------------------------------------------//
//data types//

//----------------------------------------------------------------//
// helper functions //
inline long long sum(long long n){
    return (n*(n-1))/2;
}

//----------------------------------------------------------------//
// solve function//
void solve()
{
    long long upperLimit,need;cin>>upperLimit>>need;
    bool isPossible = false;
    long long one,One;
    for(long long i=0;i<=(upperLimit+1)/2;i++){
        if(sum(i)+sum(upperLimit-i)==need){
            isPossible=true;
            one=i;
            One=upperLimit-i;
            break;
        }
    }
    if(isPossible){
        cout<<"YES"<<endl;
        while(one--)cout<<-1<<' ';
        while(One--)cout<<1<<' ';
        cout<<endl;
    }
    else cout<<"NO"<<endl;
}

//----------------------------------------------------------------//
// main function//
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    long long test = 1;
    cin >> test;
    while (test--)
    {
        solve();
    }
}