#line 2 "cpplib/util/template.hpp" #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") #pragma GCC target("avx2") #include using namespace std; #line 1 "cpplib/util/ioutil.hpp" // template // std::ostream& output(std::ostream& out,const Head& head,const Args&... args){ // out>>head; // return output(head,args...); // } // template // std::ostream& output(std::ostream& out,const Head& head){ // out>>head; // return out; // } template std::ostream& operator<<(std::ostream& out,std::pairv){ out<<"("< // ostream& operator<<(ostream& out,std::tuplev){ // std::apply(output,v); // return out; // } #line 8 "cpplib/util/template.hpp" struct __INIT__{__INIT__(){cin.tie(0);ios::sync_with_stdio(false);cout< vec; typedef vector> mat; typedef vector>> mat3; typedef vector svec; typedef vector> smat; templateusing V=vector; templateusing VV=V>; templateinline void output(T t){bool f=0;for(auto i:t){cout<<(f?" ":"")<inline void output2(T t){for(auto i:t)output(i);} templateinline void debug(T t){bool f=0;for(auto i:t){cerr<<(f?" ":"")<inline void debug2(T t){for(auto i:t)debug(i);} #define loop(n) for(long long _=0;_<(long long)(n);++_) #define _overload4(_1,_2,_3,_4,name,...) name #define __rep(i,a) repi(i,0,a,1) #define _rep(i,a,b) repi(i,a,b,1) #define repi(i,a,b,c) for(long long i=(long long)(a);i<(long long)(b);i+=c) #define rep(...) _overload4(__VA_ARGS__,repi,_rep,__rep)(__VA_ARGS__) #define _overload3_rev(_1,_2,_3,name,...) name #define _rep_rev(i,a) repi_rev(i,0,a) #define repi_rev(i,a,b) for(long long i=(long long)(b)-1;i>=(long long)(a);--i) #define rrep(...) _overload3_rev(__VA_ARGS__,repi_rev,_rep_rev)(__VA_ARGS__) #define all(n) begin(n),end(n) templatebool chmin(T& s,const E& t){bool res=s>t;s=min(s,t);return res;} templatebool chmax(T& s,const E& t){bool res=s(s,t);return res;} const vector dx={1,0,-1,0,1,1,-1,-1}; const vector dy={0,1,0,-1,1,-1,1,-1}; #define SUM(v) accumulate(all(v),0LL) #if __cplusplus>=201703L templateauto make_vector(T x,int arg,Args ...args){if constexpr(sizeof...(args)==0)return vector(arg,x);else return vector(arg,make_vector(x,args...));} #endif #define extrep(v,...) for(auto v:__MAKE_MAT__({__VA_ARGS__})) #define bit(n,a) ((n>>a)&1) vector> __MAKE_MAT__(vector v){if(v.empty())return vector>(1,vector());long long n=v.back();v.pop_back();vector> ret;vector> tmp=__MAKE_MAT__(v);for(auto e:tmp)for(long long i=0;i>; templateusing graph_w=vector>>; #if __cplusplus>=201703L constexpr inline long long powll(long long a,long long b){long long res=1;while(b--)res*=a;return res;} #endif templatepair& operator+=(pair&s,const pair&t){s.first+=t.first;s.second+=t.second;return s;} templatepair& operator-=(pair&s,const pair&t){s.first-=t.first;s.second-=t.second;return s;} templatepair operator+(const pair&s,const pair&t){auto res=s;return res+=t;} templatepair operator-(const pair&s,const pair&t){auto res=s;return res-=t;} #define BEGIN_STACK_EXTEND(size) void * stack_extend_memory_ = malloc(size);void * stack_extend_origin_memory_;char * stack_extend_dummy_memory_ = (char*)alloca((1+(int)(((long long)stack_extend_memory_)&127))*16);*stack_extend_dummy_memory_ = 0;asm volatile("mov %%rsp, %%rbx\nmov %%rax, %%rsp":"=b"(stack_extend_origin_memory_):"a"((char*)stack_extend_memory_+(size)-1024)); #define END_STACK_EXTEND asm volatile("mov %%rax, %%rsp"::"a"(stack_extend_origin_memory_));free(stack_extend_memory_); #line 2 "main.cpp" int main(){ lint n,k; cin>>n>>k; lint t=n*(n+1)/2; if(t%k){ cout<<"No"<,vector>,greater>>que; rep(i,k){ que.emplace(0,i); } if(n/k%2&&k%2&&k*3<=n){ rrep(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{ rrep(i,1,n+1){ auto d=que.top(); que.pop(); ans[d.second].emplace_back(i); que.emplace(d.first+i,d.second); } sets; rep(i,k)if((lint)ans[i].size()!=n/k){ cout<<"No"<>n>>k; // vec v(n); // rep(i,n)cin>>v[i]; // sort(all(v)); // auto get=[&](int to,int from){ // return (to-from)*v[to-1]; // }; // cout<(n, k,INF, get)< table(lint n){ // lint l = 1,r; // mapres; // res[1]=n; // while(l <= n){ // r = n / (n / l) + 1; // res[r]=n/r-n/l; // l = r; // } // return res; // } // map merge(maps,mapt){ // mapres; // for(auto [p,q]:s)res[p]+=q; // for(auto [p,q]:t)res[p]-=q; // lint pre=0; // for(auto [p,q]:res){ // res[p]+=pre; // pre=res[p]; // } // return res; // } // mapget_mebius(lint n){ // Dirichlet::resize(n); // auto dir=Dirichlet::mebius(); // vectorsuma=dir.a; // rep(i,suma.size()-1)suma[i+1]+=suma[i]; // auto get=[&](long long k){return (k<=Dirichlet::K?suma[k]:dir.A[n/k]);}; // lint l = 1,r; // mapres; // // debug(suma); // res[1]=get(1); // while(l <= n){ // r = n / (n / l) + 1; // res[r]=get(r); // l = r; // } // return res; // } // map merge2(maps,mapt){ // mapres; // for(auto [p,q]:s)res[p]=q; // for(auto [p,q]:t)res[p]=q; // return res; // } // void solve(){ // lint l,r; // cin>>l>>r; // vec v(r+1); // auto tb=merge(table(r),table(l-1)); // lint pre=0; // lint preq=0; // auto res=merge2(get_mebius(r),l!=1?get_mebius(l-1):map()); // mint ans=0; // for(auto [p,q]:tb){ // // cerr<>t; // while(t--)solve(); // }