CSES – Two Sets

數學

#include<bits/stdc++.h>
#define int long long
#define pii pair<int,int>
#define f first
#define s second
#define fastio ios_base::sync_with_stdio(0);cin.tie(0)
using namespace std;
//declare
int n;
vector<int> v1,v2;
//
signed main(){
    fastio;
    cin>>n;
    int check = (n+1) * n / 2;
    if(check % 2) cout<<"NO\n";
    else{
        check /= 2;
        int sum = 0;
        for(int i=n;i>0;i--){
            if(sum + i <= check){
                v1.push_back(i);
                sum += i;
            }
            else v2.push_back(i);
        }
        cout<<"YES\n";
        cout<<v1.size()<<"\n";
        for(int i:v1) cout<<i<<" ";
        cout<<"\n";
        cout<<v2.size()<<"\n";
        for(int i:v2) cout<<i<<" ";
        cout<<"\n";
    }
    return 0;
}

相關文章

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *