Advertisement
MathQ_

Untitled

Aug 24th, 2021
143
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.67 KB | None | 0 0
  1. #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx")
  2. #pragma GCC optimize 03
  3. #pragma GCC optimize("unroll-loops")
  4.  
  5. #include <bits/stdc++.h>
  6. #include <ext/pb_ds/assoc_container.hpp>
  7. #include <ext/pb_ds/tree_policy.hpp>
  8.  
  9. using namespace std;
  10. using namespace __gnu_pbds;
  11.  
  12. typedef unsigned long long ull;
  13. typedef long long ll;
  14. typedef long double ld;
  15. typedef std::pair<int, int> pii;
  16. typedef std::pair<char, char> pcc;
  17. typedef std::pair<ll, ll> pll;
  18. typedef tree<int,
  19.              null_type,
  20.              less<int>,
  21.              rb_tree_tag,
  22.              tree_order_statistics_node_update> ordered_set;
  23.  
  24. #define fast ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
  25. #define file_in freopen("input.txt", "r", stdin);
  26. #define file_in_out freopen("file.in", "r", stdin); freopen("file.out", "w", stdout);
  27. #define all(x) (x).begin(), (x).end()
  28. #define sz(x) (int)x.size()
  29. #define fi first
  30. #define se second
  31.  
  32. template<typename T>
  33. istream& operator>>(istream &in, vector<T> &v) {
  34.     for (auto &it : v) {
  35.         in >> it;
  36.     }
  37.     return in;
  38. }
  39.  
  40. template<typename T>
  41. ostream& operator<<(ostream &out, const vector<T> &v) {
  42.     for (auto &it : v) {
  43.         out << it << " ";
  44.     }
  45.     return out;
  46. }
  47.  
  48. template<typename T1, typename T2>
  49. istream& operator>>(istream &in, pair<T1, T2> &v) {
  50.     in >> v.fi >> v.se;
  51.     return in;
  52. }
  53.  
  54. template<typename T1, typename T2>
  55. ostream& operator<<(ostream &out, const pair<T1, T2> &v) {
  56.     out << v.fi << " " << v.se;
  57.     return out;
  58. }
  59.  
  60. vector<vector<int>> g;
  61. vector<int> mt;
  62. vector<char> used;
  63.  
  64. void resize_all(int n, int m) {
  65.     g.resize(n);
  66.     mt.resize(m, -1);
  67.     used.resize(n);
  68. }
  69.  
  70. bool dfs(int v) {
  71.     if (used[v]) return false;
  72.     used[v] = true;
  73.     for (auto u : g[v]) {
  74.         if (mt[u] == -1 || dfs(mt[u])) {
  75.             mt[u] = v;
  76.             return true;
  77.         }
  78.     }
  79.     return false;
  80. }
  81.  
  82. int main() {
  83.     fast
  84.     // file_in
  85.  
  86.     int m;
  87.     cin >> m;
  88.     string s;
  89.     cin >> s;
  90.     vector<string> cub(m);
  91.     cin >> cub;
  92.     int n = sz(s);
  93.     resize_all(m, n);
  94.     for (int i = 0; i < n; ++i) {
  95.         for (int j = 0; j < m; ++j) {
  96.             if (find(all(cub[j]), s[i]) != s.end()) {
  97.                 g[j].push_back(i);
  98.             }
  99.         }
  100.     }
  101.     for (int i = 0; i < m; ++i) {
  102.         if (dfs(i)) {
  103.             for (auto &el : used) el = 0;
  104.         }
  105.     }
  106.     vector<int> mts;
  107.     for (int i = 0; i < n; ++i) {
  108.         if (mt[i] != -1) {
  109.             mts.push_back(mt[i] + 1);
  110.         }
  111.     }
  112.     if (sz(mts) != n) {
  113.         cout << "NO" << '\n';
  114.         return 0;
  115.     }
  116.     cout << "YES" << '\n';
  117.     cout << mts << '\n';
  118.     return 0;
  119. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement