結果
問題 | No.942 プレゼント配り |
ユーザー |
![]() |
提出日時 | 2020-08-14 00:12:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,728 bytes |
コンパイル時間 | 2,364 ms |
コンパイル使用メモリ | 203,496 KB |
最終ジャッジ日時 | 2025-01-12 22:05:44 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 RE * 1 |
other | AC * 11 RE * 7 |
ソースコード
#include <bits/stdc++.h> #define rep(i, a) for (int i = (int)0; i < (int)a; ++i) #define rrep(i, a) for (int i = (int)a - 1; i >= 0; --i) #define REP(i, a, b) for (int i = (int)a; i < (int)b; ++i) #define RREP(i, a, b) for (int i = (int)a - 1; i >= b; --i) #define pb push_back #define eb emplace_back #define all(x) x.begin(), x.end() #define rall(x) x.rbegin(), x.rend() #define popcount __builtin_popcount using ll = long long; constexpr ll mod = 1e9 + 7; constexpr ll INF = 1LL << 60; template <class T> inline bool chmin(T &a, T b) { if (a > b) { a = b; return true; } return false; } template <class T> inline bool chmax(T &a, T b) { if (a < b) { a = b; return true; } return false; } ll gcd(ll n, ll m) { ll tmp; while (m != 0) { tmp = n % m; n = m; m = tmp; } return n; } ll lcm(ll n, ll m) { return abs(n) / gcd(n, m) * abs(m); //gl=xy } using namespace std; void solve() { ll n,k; cin>>n>>k; if(k==1){ cout<<"Yes\n"; rep(i,n){ if(i)cout<<" "; cout<<i+1; } cout<<"\n"; return; } ll sum=(n+1)*n/2; if(sum%k==0&&n!=k){ set<ll>st; cout<<"Yes\n"; vector<vector<ll>>v(n/k); rep(i,n){ v[i/k].pb(i+1); st.emplace(i+1); } rep(i,k){ int m=i; ll res=0; if(n%(2*k)==0) rep(j,n/(2*k)){ if(j)cout<<" "; cout<<v[j][m]<<" "; cout<<v[n/k-1-j][k-1-m]; m=(m+1)%k; } else{ rep(j,n/k){ if(j)cout<<" "; if(j!=n/k-1)cout<<v[j][m],res+=v[j][m],st.erase(v[j][m]); else cout<<sum/k-res,st.erase(sum/k-res); m=(m+2)%k; } } cout<<"\n"; } assert(st.size()==0); } else cout<<"No\n"; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout << fixed << setprecision(15); solve(); return 0; }