#894726 (C++17) No.1123 Afforestation

提出ソース

結果

問題	No.1123 Afforestation
ユーザー	snuke
提出日時	2023-07-23 02:38:45
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	1,157 ms / 2,500 ms
コード長	9,026 bytes
コンパイル時間	2,815 ms
コンパイル使用メモリ	229,016 KB
実行使用メモリ	117,392 KB
最終ジャッジ日時	2024-09-23 01:49:30
合計ジャッジ時間	19,285 ms
ジャッジサーバーID （参考情報）	judge5 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	3 ms 5,376 KB
testcase_04	AC	11 ms 6,440 KB
testcase_05	AC	113 ms 32,368 KB
testcase_06	AC	515 ms 117,264 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	3 ms 5,376 KB
testcase_11	AC	10 ms 5,376 KB
testcase_12	AC	91 ms 14,404 KB
testcase_13	AC	1,110 ms 117,012 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	3 ms 5,376 KB
testcase_19	AC	18 ms 6,324 KB
testcase_20	AC	395 ms 31,968 KB
testcase_21	AC	1,066 ms 116,820 KB
testcase_22	AC	1,157 ms 116,956 KB
testcase_23	AC	1,077 ms 117,268 KB
testcase_24	AC	969 ms 117,144 KB
testcase_25	AC	1,003 ms 117,268 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	316 ms 23,296 KB
testcase_29	AC	2 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	321 ms 23,040 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	3 ms 5,376 KB
testcase_35	AC	3 ms 5,376 KB
testcase_36	AC	2 ms 5,376 KB
testcase_37	AC	2 ms 5,376 KB
testcase_38	AC	2 ms 5,376 KB
testcase_39	AC	4 ms 5,376 KB
testcase_40	AC	4 ms 5,376 KB
testcase_41	AC	3 ms 5,376 KB
testcase_42	AC	9 ms 5,376 KB
testcase_43	AC	18 ms 6,324 KB
testcase_44	AC	38 ms 9,444 KB
testcase_45	AC	38 ms 9,324 KB
testcase_46	AC	46 ms 10,464 KB
testcase_47	AC	45 ms 10,720 KB
testcase_48	AC	2 ms 5,376 KB
testcase_49	AC	2 ms 5,376 KB
testcase_50	AC	4 ms 5,376 KB
testcase_51	AC	3 ms 5,376 KB
testcase_52	AC	2 ms 5,376 KB
testcase_53	AC	2 ms 5,376 KB
testcase_54	AC	2 ms 5,376 KB
testcase_55	AC	4 ms 5,376 KB
testcase_56	AC	4 ms 5,376 KB
testcase_57	AC	3 ms 5,376 KB
testcase_58	AC	9 ms 5,376 KB
testcase_59	AC	18 ms 6,328 KB
testcase_60	AC	36 ms 9,320 KB
testcase_61	AC	37 ms 9,448 KB
testcase_62	AC	44 ms 10,656 KB
testcase_63	AC	47 ms 10,584 KB
testcase_64	AC	2 ms 5,376 KB
testcase_65	AC	2 ms 5,376 KB
testcase_66	AC	2 ms 5,376 KB
testcase_67	AC	2 ms 5,376 KB
testcase_68	AC	4 ms 5,376 KB
testcase_69	AC	5 ms 5,376 KB
testcase_70	AC	8 ms 5,376 KB
testcase_71	AC	18 ms 6,452 KB
testcase_72	AC	2 ms 5,376 KB
testcase_73	AC	2 ms 5,376 KB
testcase_74	AC	2 ms 5,376 KB
testcase_75	AC	2 ms 5,376 KB
testcase_76	AC	2 ms 5,376 KB
testcase_77	AC	2 ms 5,376 KB
testcase_78	AC	2 ms 5,376 KB
testcase_79	AC	2 ms 5,376 KB
testcase_80	AC	2 ms 5,376 KB
testcase_81	AC	2 ms 5,376 KB
testcase_82	AC	2 ms 5,376 KB
testcase_83	AC	2 ms 5,376 KB
testcase_84	AC	234 ms 117,264 KB
testcase_85	AC	259 ms 117,268 KB
testcase_86	AC	2 ms 5,376 KB
testcase_87	AC	2 ms 5,376 KB
testcase_88	AC	218 ms 117,076 KB
testcase_89	AC	218 ms 117,392 KB
testcase_90	AC	2 ms 5,376 KB
testcase_91	AC	2 ms 5,376 KB
testcase_92	AC	2 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define rep1(i,n) for(int i = 1; i <= (n); ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
#define srep(i,s,t) for (int i = s; i < (t); ++i)
#define rng(a) a.begin(),a.end()
#define rrng(a) a.rbegin(),a.rend()
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define em emplace
#define pob pop_back
#define sz(x) (int)(x).size()
#define pcnt __builtin_popcountll
#define snuke srand((unsigned)clock()+(unsigned)time(NULL));
#define newline puts("")
#define vc vector
using namespace std;
template<class T> using vv = vc<vc<T>>;
template<class T> using PQ = priority_queue<T,vc<T>,greater<T>>;
using uint = unsigned; using ull = unsigned long long;
using vi = vc<int>; using vvi = vv<int>; using vvvi = vv<vi>;
using ll = long long; using vl = vc<ll>; using vvl = vv<ll>; using vvvl = vv<vl>;
using P = pair<int,int>; using vp = vc<P>; using vvp = vv<P>;
int geti(){int x;scanf("%d",&x);return x;}
vi pm(int n, int s=0) { vi a(n); iota(rng(a),s); return a;}
template<class T1,class T2>istream& operator>>(istream&i,pair<T1,T2>&v){return i>>v.fi>>v.se;}
template<class T1,class T2>ostream& operator<<(ostream&o,const pair<T1,T2>&v){return o<<v.fi<<","<<v.se;}
template<class T>istream& operator>>(istream&i,vc<T>&v){rep(j,sz(v))i>>v[j];return i;}
template<class T>string join(const T&v,const string& d=""){stringstream s;rep(i,sz(v))(i?s<<d:s)<<v[i];return s.str();}
template<class T>ostream& operator<<(ostream&o,const vc<T>&v){if(sz(v))o<<join(v," ");return o;}
template<class T>void vin(vv<T>& a){int n,m;cin>>n>>m;a=vv<T>(n,vc<T>(m));cin>>a;}
template<class T>void vin(vc<T>& a){int n;cin>>n;a=vc<T>(n);cin>>a;}
template<class T1,class T2>void operator--(pair<T1,T2>&a,int){a.fi--;a.se--;}
template<class T1,class T2>void operator++(pair<T1,T2>&a,int){a.fi++;a.se++;}
template<class T>void operator--(vc<T>&a,int){for(T&x:a)x--;}
template<class T>void operator++(vc<T>&a,int){for(T&x:a)x++;}
template<class T>void operator+=(vc<T>&a,T b){for(T&x:a)x+=b;}
template<class T>void operator-=(vc<T>&a,T b){for(T&x:a)x-=b;}
template<class T>void operator*=(vc<T>&a,T b){for(T&x:a)x*=b;}
template<class T>void operator/=(vc<T>&a,T b){for(T&x:a)x/=b;}
template<class T>void operator+=(vc<T>&a,const vc<T>&b){a.insert(a.end(),rng(b));}
template<class T1,class T2>bool mins(T1& x,const T2&y){if(y<x){x=y;return true;}else return false;}
template<class T1,class T2>bool maxs(T1& x,const T2&y){if(x<y){x=y;return true;}else return false;}
template<class T>T min(const vector<T>& a){return *min_element(rng(a));}
template<class T>T max(const vector<T>& a){return *max_element(rng(a));}
template<class Tx,class Ty>Tx dup(Tx x, Ty y){return (x+y-1)/y;}
template<class T>ll suma(const vc<T>&a){ll res(0);for(auto&&x:a)res+=x;return res;}
template<class T>ll suma(const vv<T>&a){ll res(0);for(auto&&x:a)res+=suma(x);return res;}
template<class T>void uni(T& a){sort(rng(a));a.erase(unique(rng(a)),a.end());}
template<class T>void prepend(vc<T>&a,const T&x){a.insert(a.begin(),x);}
const double eps = 1e-10;
const ll LINF = 1001002003004005006ll;
const int INF = 1001001001;
#define dame { puts(":("); return 0;}
#define yes { puts("Yes"); return 0;}
#define no { puts("No"); return 0;}
#define rtn(x) { cout<<(x)<<endl; return 0;}
#define yn {puts("Yes");}else{puts("No");}
const int MX = 200005;
using vs = vc<string>;
void vin(vs& s){int n,m;cin>>n>>m;s=vs(n);cin>>s;}


// Max flow
// !! Be care of double and INF !!
struct Maxflow {
  typedef int TT;
  int n;
  vi to, next, head, dist, it;
  vector<TT> lim;
  Maxflow(){}
  Maxflow(int n):n(n),head(n,-1),it(n){}
  void add(int a, int b, TT c=1) {
    next.pb(head[a]); head[a] = sz(to); to.pb(b); lim.pb(c);
    next.pb(head[b]); head[b] = sz(to); to.pb(a); lim.pb(0); 
  }
  void bfs(int sv) {
    dist = vi(n,INF);
    queue<int> q;
    dist[sv] = 0; q.push(sv);
    while (!q.empty()){
      int v = q.front(); q.pop();
      for (int i = head[v]; i != -1; i = next[i]) {
        if (lim[i] && dist[to[i]] == INF) { // double !!
          dist[to[i]] = dist[v]+1; q.push(to[i]);
        }
      }
    }
  }
  TT dfs(int v, int tv, TT nf=INF) { // INF !!
    if (v == tv) return nf;
    for (; it[v] != -1; it[v] = next[it[v]]) {
      int u = to[it[v]]; TT f;
      if (!lim[it[v]] || dist[v] >= dist[u]) continue; // double !!
      if (f = dfs(u, tv, min(nf, lim[it[v]])), f) { // double !!
        lim[it[v]] -= f;
        lim[it[v]^1] += f;
        return f;
      }
    }
    return 0;
  }
  TT solve(int sv, int tv) {
    TT flow = 0, f;
    while (1) {
      bfs(sv);
      if (dist[tv] == INF) return flow;
      rep(i,n) it[i] = head[i];
      while (f = dfs(sv,tv), f) flow += f;
    }
  }
};
//

// coordinate compression
struct X {
  typedef int T;
  vector<T> d;
  X() {}
  // X(vector<T>& a): d(a) {
  //   init();
  //   for (T& na : a) na = (*this)(na);
  // }
  void add(const T& x) { d.pb(x);}
  void init() {
    sort(rng(d));
    d.erase(unique(rng(d)), d.end());
  }
  int size() const { return sz(d);}
  T operator[](int i) const { return d[i];}
  int operator()(const T& x) const { return upper_bound(rng(d),x)-d.begin()-1;}
  bool in(const T& x) const { return binary_search(rng(d), x);}
};
//



int main() {
  int h,w;
  scanf("%d%d",&h,&w);
  vi a(h), b(w);
  cin>>a>>b;
  int n;
  scanf("%d",&n);
  vp p(n); cin>>p;
  rep(i,n) p[i].fi--;
  rep(i,n) p[i].se--;


  // X xs, ys;
  // rep(i,n) xs.add(p[i].fi);
  // rep(i,n) ys.add(p[i].se);
  // xs.init();
  // ys.init();
  // int eh = sz(xs), dh = h-eh;
  // int ew = sz(ys), dw = w-ew;

  vs ans(h,string(w,'.'));
  rep(i,n) ans[p[i].fi][p[i].se] = 'x';

  // int sv = h+w, tv = sv+1;
  // Maxflow g(tv+1);
  // rep(ei,eh) {
  //   int i = xs[ei];
  //   rep(j,w) {
  //     if (ys.in(j)) continue;
  //     g.add(i,h+j);
  //   }
  // }
  // rep(ej,ew) {
  //   int j = ys[ej];
  //   rep(i,h) {
  //     if (xs.in(i)) continue;
  //     g.add(i,h+j);
  //   }
  // }

  // rep(i,h) {
  //   if (xs.in(i)) g.add(sv,i,a[i]);
  //   else {
  //     int now = max(0,a[i]-dw);
  //     g.add(sv,i,now);
  //   }
  // }
  // rep(j,w) {
  //   if (ys.in(j)) g.add(h+j,tv,b[j]);
  //   else {
  //     int now = max(0,b[j]-dh);
  //     g.add(h+j,tv,now);
  //   }
  // }
  // g.solve(sv,tv);


  // vi ta = a, tb = b;
  // int e = 0;
  // rep(ei,eh) {
  //   int i = xs[ei];
  //   rep(j,w) {
  //     if (ys.in(j)) continue;
  //     if (!g.lim[e]) tb[j]--;
  //     e += 2;
  //   }
  // }
  // rep(ej,ew) {
  //   int j = ys[ej];
  //   rep(i,h) {
  //     if (xs.in(i)) continue;
  //     if (!g.lim[e]) ta[i]--;
  //     e += 2;
  //   }
  // }
  // rep(i,h) if (!xs.in(i) && ta[i] > dw) dame;
  // rep(j,w) if (!ys.in(j) && tb[j] > dh) dame;

  // int ne = sz(g.lim);
  // rep(ei,eh) {
  //   int i = xs[ei];
  //   rep(ej,ew) {
  //     int j = ys[ej];
  //     if (ans[i][j] == 'x') continue;
  //     g.add(i,h+j);
  //   }
  // }
  // rep(i,h) {
  //   if (xs.in(i)) continue;
  //   g.add(sv,i, min(a[i],dw));
  // }
  // rep(j,w) {
  //   if (ys.in(j)) continue;
  //   g.add(h+j,tv, min(b[j],dh));
  // }
  // g.solve(sv,tv);

  auto add = [&](int i, int j) {
    ans[i][j] = 'o';
    a[i]--;
    b[j]--;
  };

  // e = 0;
  // rep(ei,eh) {
  //   int i = xs[ei];
  //   rep(j,w) {
  //     if (ys.in(j)) continue;
  //     if (!g.lim[e]) add(i,j);
  //     e += 2;
  //   }
  // }
  // rep(ej,ew) {
  //   int j = ys[ej];
  //   rep(i,h) {
  //     if (xs.in(i)) continue;
  //     if (!g.lim[e]) add(i,j);
  //     e += 2;
  //   }
  // }
  // e = ne;
  // rep(ei,eh) {
  //   int i = xs[ei];
  //   rep(ej,ew) {
  //     int j = ys[ej];
  //     if (ans[i][j] == 'x') continue;
  //     if (!g.lim[e]) add(i,j);
  //     e += 2;
  //   }
  // }
  // rep(ei,eh) if (a[xs[ei]]) dame;
  // rep(ej,ew) if (b[ys[ej]]) dame;



  // vi is, js;
  // rep(i,h) if (!xs.in(i)) is.pb(i);
  // rep(j,w) if (!ys.in(j)) js.pb(j);
  // for (int i : is) {
  //   vp nj;
  //   for (int j : js) nj.eb(b[j],j);
  //   sort(rrng(nj));
  //   int na = a[i];
  //   rep(k,na) add(i,nj[k].se);
  // }

  int sv = h+w, tv = sv+1;
  Maxflow g(tv+1);

  vi oa = a, ob = b;

  vvi ng(h,vi(w));
  rep(i,h) {
    vp nj;
    rep(j,w) nj.eb(b[j],j);
    sort(rrng(nj));
    rep(k,a[i]) {
      int j = nj[k].se;
      ng[i][j] = 1;
      b[j]--;
    }
    a[i] = 0;
  }
  rep(j,w) if (b[j]) dame;
  int e = 0;
  rep(i,h)rep(j,w) {
    if (ans[i][j] == 'x') {
      if (!ng[i][j]) continue;
      a[i]++; b[j]++;
    } else {
      g.add(i,h+j);
      if (ng[i][j]) {
        g.lim[e] = 0;
        g.lim[e^1] = 1;
      }
      e += 2;
    }
  }
  rep(i,h) if (a[i]) g.add(sv,i,a[i]);
  rep(j,w) if (b[j]) g.add(h+j,tv,b[j]);
  g.solve(sv,tv);
  a = oa; b = ob;
  e = 0;
  rep(i,h)rep(j,w) {
    if (ans[i][j] == 'x') continue;
    if (!g.lim[e]) add(i,j);
    e += 2;
  }

  rep(i,h) if (a[i]) dame;
  rep(j,w) if (b[j]) dame;

  cout<<"Yay!"<<endl;
  rep(i,h) cout<<ans[i]<<endl;
  return 0;
}

yukicoder

結果

テストケース

ソースコード