結果

問題 No.942 プレゼント配り
コンテスト
ユーザー hotman78
提出日時 2019-12-05 15:09:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 6,476 bytes
コンパイル時間 2,804 ms
コンパイル使用メモリ 206,772 KB
最終ジャッジ日時 2024-11-14 21:55:36
合計ジャッジ時間 3,219 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp:87:9: error: non-local lambda expression cannot have a capture-default
   87 | auto Y=[&](auto f){return[&](auto... args){return f(f, args...);};};
      |         ^
main.cpp:93:19: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   93 | inline bool chmin(auto& s,const auto& t){bool res=s>t;s=min(s,t);return res;}
      |                   ^~~~
main.cpp:93:33: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   93 | inline bool chmin(auto& s,const auto& t){bool res=s>t;s=min(s,t);return res;}
      |                                 ^~~~
main.cpp:94:19: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   94 | inline bool chmax(auto& s,const auto& t){bool res=s<t;s=max(s,t);return res;}
      |                   ^~~~
main.cpp:94:33: warning: use of 'auto' in parameter declaration only available with '-std=c++20' or '-fconcepts'
   94 | inline bool chmax(auto& s,const auto& t){bool res=s<t;s=max(s,t);return res;}
      |                                 ^~~~

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target ("avx")
#include<bits/stdc++.h>
using namespace::std;
//struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}} __init;
__attribute__((constructor))
void init(){
	cin.tie(0);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(15);
}
// random_device rd;
// mt19937 mt(rd());
// #include <boost/numeric/interval.hpp>
// #include <boost/numeric/interval/io.hpp>
// #include <boost/intrusive/rbtree.hpp>
// #include <boost/dynamic_bitset.hpp>
// #include <boost/geometry.hpp>
// #include <boost/geometry/geometries/linestring.hpp>
// #include <boost/geometry/geometries/polygon.hpp>
// #include <boost/geometry/geometries/point_xy.hpp>
// #include <boost/math/tools/minima.hpp>
// using boost::math::tools::brent_find_minima;
// #include <boost/multiprecision/cpp_dec_float.hpp>
// #include <boost/multiprecision/cpp_int.hpp>
// namespace mp = boost::multiprecision;
// typedef mp::number<mp::cpp_dec_float<256>> cdouble;
// typedef mp::cpp_int cint;
// #include <boost/unordered_map.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tag_and_trait.hpp>
template<typename T=long long>using pbds=__gnu_pbds::tree<T,__gnu_pbds::null_type,less<T>,__gnu_pbds::rb_tree_tag,__gnu_pbds::tree_order_statistics_node_update>;
template<typename T=long long>using pbds_map=__gnu_pbds::tree<T,T,less<T>,__gnu_pbds::rb_tree_tag,__gnu_pbds::tree_order_statistics_node_update>;
template<typename T=long long>using pqueue =__gnu_pbds::priority_queue<T, less<T>,__gnu_pbds::rc_binomial_heap_tag>;
typedef long long lint;
typedef long long ll;
typedef long double ldouble;
typedef vector<lint> vec;
typedef vector<vector<lint>> mat;
typedef vector<vector<vector<lint>>> mat3;
typedef vector<double> dvec;
typedef vector<vector<double>> dmat;
typedef vector<vector<vector<double>>> dmat3;
typedef vector<string> svec;
typedef vector<vector<string>> smat;
typedef vector<vector<vector<string>>> smat3;
template<typename T>using Vec=vector<T>;
template<typename T>using Mat=vector<vector<T>>;
template<typename T>using Mat3=vector<vector<vector<T>>>;
template<typename T>using Mat4=vector<vector<vector<vector<T>>>>;
template<typename S,typename T>using pvec=vector<pair<S,T>>;
template<typename S,typename T>using pmat=vector<vector<pair<S,T>>>;
template<typename S,typename T>using pmat3=vector<vector<vector<pair<S,T>>>>;
template<typename S,typename T>using pmat4=vector<vector<vector<vector<pair<S,T>>>>>;
template<typename... T>using tvec=vector<tuple<T...>>;
template<typename... T>using tmat=vector<vector<tuple<T...>>>;
template<typename... T>using tmat3=vector<vector<vector<tuple<T...>>>>;
template<typename... T>using tmat4=vector<vector<vector<vector<tuple<T...>>>>>;
#define rep(i, n) for(lint i = 0; i < (lint)(n); i++)
#define irep(i) for(lint i = 0;; i++)
#define irep1(i) for(lint i = 1;; i++)
#define irep2(i) for(lint i = 2;; i++)
#define rrep(i, n) for(lint i = (lint)(n-1); i >-1; i--)
#define rrepi(i,a,b) for(lint i = (lint)(b-1); i >a-1; i--)
#define repi(i,a,b) for(lint i=lint(a);i<lint(b);i++)
#define rep2(i,a,b,c) for(lint i=lint(a);i>lint(b);i+=c)
#define all(x) (x).begin(),(x).end()
#define PI 3.141592653589793
#define dist(x1,y1,x2,y2) (pow(pow(x2-x1,2)+pow(y2-y1,2),0.5))
//#define input(a,n) lint n;cin>>n;vector<lint>a(n);rep(i,n)cin>>a[i];
#define SUM(v) accumulate(all(v),0LL)
#define INF (1LL<<60)
#define IINF (1<<30)
#define EPS (1e-10)
#define LINF 9223372036854775807
//#define MOD 998244353LL
#define MOD 1000000007LL
#define endl "\n"
template<typename T>inline void numout(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<<i<INF/2?i:"INF";f=1;}cout<<endl;}
template<typename T>inline void numout2(T t){for(auto i:t)numout(i);}
template<typename T>inline void output(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<<i;f=1;}cout<<endl;}
template<typename T>inline void output2(T t){for(auto i:t)output(i);}
template<typename T>inline void _output(T t){bool f=0;for(int i=0;i<t.size();i++){cout<<f?"":" "<<t[i];f=1;}cout<<endl;}
template<typename T>inline void _output2(T t){for(int i=0;i<t.size();i++)output(t[i]);}
auto Y=[&](auto f){return[&](auto... args){return f(f, args...);};};
template<typename T=lint>T in(){return *istream_iterator<T>(cin);}
template<typename T=lint>inline T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<typename T=lint>inline T lcm(T a,T b){return a/gcd(a,b)*b;}
template<typename T>inline T minq(T a,T b){return min(a,b);}
template<typename T>inline T maxq(T a,T b){return max(a,b);}
inline bool chmin(auto& s,const auto& t){bool res=s>t;s=min(s,t);return res;}
inline bool chmax(auto& s,const auto& t){bool res=s<t;s=max(s,t);return res;}
const vector<lint> dx={-1,1,0,0,1,1,-1,-1};
const vector<lint> dy={0,0,-1,1,1,-1,1,-1};

int main(){
    lint n,k;
    cin>>n>>k;
    lint t=n*(n+1)/2;
    if(t%k){
        cout<<"No"<<endl;
        return 0;
    }
    if(n%k){
        cout<<"No"<<endl;
        return 0;
    }
    t/=k;
    mat ans(k);
    vec su(k);
    priority_queue<pair<lint,lint>,vector<pair<lint,lint>>,greater<pair<lint,lint>>>que;
    rep(i,k){
        que.emplace(0,i);
    }
    if(n/k%2&&k%2&&k*3<=n){
        rrepi(i,k*3+1,n+1){
            auto d=que.top();
            que.pop();
            ans[d.second].emplace_back(i);
            que.emplace(d.first+i,d.second);
        }
        mat v(k,vec(3));
        rep(i,k){
            v[i][0]=i+1;
            v[i][1]=k+(i+k/2+1)%k+1;
            v[i][2]=(3*(k*3+1)/2)-v[i][0]-v[i][1];
            ans[i].emplace_back(v[i][0]);
            ans[i].emplace_back(v[i][1]);
            ans[i].emplace_back(v[i][2]);
        }
    }else{
        rrepi(i,1,n+1){
            auto d=que.top();
            que.pop();
            ans[d.second].emplace_back(i);
            que.emplace(d.first+i,d.second);
        }
        set<lint>s;
        rep(i,k)if((lint)ans[i].size()!=n/k){
            cout<<"No"<<endl;
            return 0;
        }
        rep(i,k)rep(j,ans[i].size()){
            if(1<=ans[i][j]&&ans[i][j]<=n)s.insert(ans[i][j]);
            else{
                cout<<"No"<<endl;
                return 0;
            }
        }
        rep(i,k)if(SUM(ans[i])!=t){
            cout<<"No"<<endl;
            return 0;
        }
        if((lint)s.size()!=n){
            cout<<"No"<<endl;
            return 0;
        }
    }
    cout<<"Yes"<<endl;
    output2(ans);
}
0