結果
問題 | No.1370 置換門松列 |
ユーザー |
![]() |
提出日時 | 2021-01-29 21:45:57 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 34 ms / 2,000 ms |
コード長 | 2,667 bytes |
コンパイル時間 | 2,402 ms |
コンパイル使用メモリ | 203,276 KB |
最終ジャッジ日時 | 2025-01-18 09:07:07 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 25 |
ソースコード
#define _USE_MATH_DEFINES #include <bits/stdc++.h> using namespace std; #define FOR(i,m,n) for(int i=(m);i<(n);++i) #define REP(i,n) FOR(i,0,n) #define ALL(v) (v).begin(),(v).end() using ll = long long; constexpr int INF = 0x3f3f3f3f; constexpr long long LINF = 0x3f3f3f3f3f3f3f3fLL; constexpr double EPS = 1e-8; constexpr int MOD = 1000000007; // constexpr int MOD = 998244353; constexpr int dy[] = {1, 0, -1, 0}, dx[] = {0, -1, 0, 1}; constexpr int dy8[] = {1, 1, 0, -1, -1, -1, 0, 1}, dx8[] = {0, -1, -1, -1, 0, 1, 1, 1}; template <typename T, typename U> inline bool chmax(T &a, U b) { return a < b ? (a = b, true) : false; } template <typename T, typename U> inline bool chmin(T &a, U b) { return a > b ? (a = b, true) : false; } struct IOSetup { IOSetup() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << fixed << setprecision(20); } } iosetup; bool is_kadomatsu(ll a, ll b, ll c) { return a != b && b != c && c != a && !(a < b && b < c) && !(a > b && b > c); } std::vector<int> topological_sort(const std::vector<std::vector<int>> &graph) { int n = graph.size(); std::vector<int> deg(n, 0); for (int i = 0; i < n; ++i) { for (int e : graph[i]) ++deg[e]; } std::queue<int> que; for (int i = 0; i < n; ++i) { if (deg[i] == 0) que.emplace(i); } std::vector<int> res; while (!que.empty()) { int ver = que.front(); que.pop(); res.emplace_back(ver); for (int e : graph[ver]) { if (--deg[e] == 0) que.emplace(e); } } return res.size() == n ? res : std::vector<int>(); } int main() { int n, m; cin >> n >> m; vector<int> a(n); REP(i, n) cin >> a[i], --a[i]; FOR(i, 2, n) { if (a[i - 2] == a[i - 1] || a[i - 1] == a[i] || a[i] == a[i - 2]) { cout << "No\n"; return 0; } } { vector<vector<int>> graph(m); for (int i = 0; i < n; i += 2) { if (i > 0) graph[a[i]].emplace_back(a[i - 1]); if (i + 1 < n) graph[a[i]].emplace_back(a[i + 1]); } vector<int> ts = topological_sort(graph); if (!ts.empty()) { vector<int> x(m); REP(i, m) x[ts[i]] = i; cout << "Yes\n"; REP(i, m) cout << x[i] + 1 << " \n"[i + 1 == m]; return 0; } } { vector<vector<int>> graph(m); for (int i = 1; i < n; i += 2) { if (i > 0) graph[a[i]].emplace_back(a[i - 1]); if (i + 1 < n) graph[a[i]].emplace_back(a[i + 1]); } vector<int> ts = topological_sort(graph); if (!ts.empty()) { vector<int> x(m); REP(i, m) x[ts[i]] = i; cout << "Yes\n"; REP(i, m) cout << x[i] + 1 << " \n"[i + 1 == m]; return 0; } } cout << "No\n"; return 0; }