#337286 (C++14) No.470 Inverse S+T Problem

提出ソース

結果

問題	No.470 Inverse S+T Problem
ユーザー	Haar
提出日時	2019-04-12 18:43:27
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	5,239 bytes
コンパイル時間	2,164 ms
コンパイル使用メモリ	193,360 KB
実行使用メモリ	171,116 KB
最終ジャッジ日時	2024-06-01 22:57:34
合計ジャッジ時間	5,602 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 13,756 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	WA	-
testcase_04	AC	1 ms 6,940 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	TLE	-
testcase_07	--	-
testcase_08	--	-
testcase_09	--	-
testcase_10	--	-
testcase_11	--	-
testcase_12	--	-
testcase_13	--	-
testcase_14	--	-
testcase_15	--	-
testcase_16	--	-
testcase_17	--	-
testcase_18	--	-
testcase_19	--	-
testcase_20	--	-
testcase_21	--	-
testcase_22	--	-
testcase_23	--	-
testcase_24	--	-
testcase_25	--	-
testcase_26	--	-
testcase_27	--	-
testcase_28	--	-
testcase_29	--	-
testcase_30	--	-

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

ソースコード

raw source code

#include <bits/stdc++.h>
#define FOR(v, a, b) for(int v = (a); v < (b); ++v)
#define FORE(v, a, b) for(int v = (a); v <= (b); ++v)
#define REP(v, n) FOR(v, 0, n)
#define REPE(v, n) FORE(v, 0, n)
#define REV(v, a, b) for(int v = (a); v >= (b); --v)
#define ALL(x) (x).begin(), (x).end()
#define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it)
#define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it)
#define EXIST(c,x) ((c).find(x) != (c).end())
#define LLI long long int
#define fst first
#define snd second

#ifdef DEBUG
#include <misc/C++/Debug.cpp>
#else
#define dump(x) ((void)0)
#endif

#define gcd __gcd

using namespace std;
template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;}

template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;}
template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;}
template <typename T, typename U> istream& operator>>(istream &is, pair<T,U> &p){is >> p.first >> p.second; return is;}

template <typename T, typename U> bool chmin(T &a, const U &b){return (a>b ? a=b, true : false);}
template <typename T, typename U> bool chmax(T &a, const U &b){return (a<b ? a=b, true : false);}
template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);}


template <typename Cost = int> class Edge{
public:
  int from,to;
  Cost cost;
  Edge() {}
  Edge(int from, int to, Cost cost): from(from), to(to), cost(cost){}

  Edge rev() const {return Edge(to,from,cost);}
  
  static bool cmp_to_lt(const Edge &e1, const Edge &e2){return e1.to < e2.to;}
  static bool cmp_cost_lt(const Edge &e1, const Edge &e2){return e1.cost < e2.cost;}
  static bool cmp_to_gt(const Edge &e1, const Edge &e2){return e1.to > e2.to;}
  static bool cmp_cost_gt(const Edge &e1, const Edge &e2){return e1.cost > e2.cost;}
  friend ostream& operator<<(ostream &os, const Edge &e){
    os << "(FROM: " << e.from << "," << "TO: " << e.to << "," << "COST: " << e.cost << ")";
    return os;
  }
};

template <typename T> using Graph = vector<vector<Edge<T>>>;

template <typename T> vector<int> strongly_connected_components(Graph<T> &graph){
  int n = graph.size();
  vector<bool> visit(n);
  vector<int> check(n);

  function<void(int)> dfs =
    [&](int cur){
      visit[cur] = true;
      for(auto &e : graph[cur]) if(!visit[e.to]) dfs(e.to);
      check.push_back(cur);
    };
  
  REP(i,n) if(!visit[i]) dfs(i);

  Graph<T> rgraph(n);
  REP(i,n) for(auto &e : graph[i]) rgraph[e.to].push_back(e.rev());

  vector<int> ret(n,-1);

  reverse(ALL(check));

  function<void(int,int)> rdfs =
    [&](int cur, int i){
      ret[cur] = i;
      for(auto &e : rgraph[cur]) if(ret[e.to] == -1) rdfs(e.to,i);
    };

  int i = 0;
  for(auto c : check) if(ret[c] == -1) {rdfs(c,i); ++i;}
    
  return ret;
}

bool tsort(const Graph<int> &graph, vector<int> &ret){
  int n = graph.size();
  vector<int> indeg(n);
  REP(i,n){
    for(auto &e : graph[i]){
      ++indeg[e.to];
    }
  }
  
  stack<int> st;
  REV(i,n-1,0){
    if(indeg[i]==0) st.push(i);
  }

  while(!st.empty()){
    int cur = st.top(); st.pop();
    ret.push_back(cur);
    for(auto &e : graph[cur]){
      --indeg[e.to];
      if(indeg[e.to]==0){
	st.push(e.to);
      }
    }
  }
  
  return (int)ret.size() == n;
}


class two_sat{
  int n;
  Graph<int> g;
  
public:
  two_sat(int n): n(n), g(2*n){}

  int inv(int i){ // not
    if(i<n) return i+n;
    else return i-n;
  }

  void add(int a, int b){
    if(a == b){ // a ∨ a <=> (a => !a)
      g[a].push_back(Edge<int>(a, inv(a), 1));
    }else{ // a ∨ b <=> (a => !b) ∧ (b => !a)
      g[a].push_back(Edge<int>(a, inv(b), 1));
      g[b].push_back(Edge<int>(b, inv(a), 1));
    }
  }
  
  bool solve(vector<bool> &ret){
    auto s = strongly_connected_components(g);
    
    REP(i,n) if(s[i] == s[i+n]) return false;

    int m = *max_element(ALL(s)) + 1;
    Graph<int> g2(m);
    vector<int> ts;

    for(auto &v : g){
      for(auto &e : v){
	if(s[e.from] != s[e.to]) g2[s[e.from]].push_back(Edge<int>(s[e.from], s[e.to], 1));
      }
    }

    tsort(g2, ts);

    vector<int> r(m);
    REP(i,m) r[ts[i]] = i;

    ret = vector<bool>(n);
    REP(i,n) ret[i] = r[s[i]] > r[s[i+n]];

    return true;
  }
};

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N;
  while(cin >> N){
    vector<string> U(N); cin >> U;

    vector<string> S1(N), T1(N), S2(N), T2(N);
    REP(i,N){
      S1[i] = {U[i][0], U[i][1]};
      T1[i] = {U[i][2]};
      S2[i] = {U[i][0]};
      T2[i] = {U[i][1], U[i][2]};
    }

    two_sat sat(N);

    REP(i,N){
      FOR(j,i+1,N){
	if(S1[i] == S1[j] or T1[i] == T1[j]) sat.add(sat.inv(i), sat.inv(j));
	if(S1[i] == T2[j] or T1[i] == S2[j]) sat.add(i, sat.inv(j));
	if(S2[i] == T1[j] or T2[i] == S1[j]) sat.add(sat.inv(i), j);
	if(S2[i] == S2[j] or T2[i] == T2[j]) sat.add(i, j);
      }
    }

    vector<bool> ret;
    bool ans = sat.solve(ret);

    if(ans){
      REP(i,N){
	if(ret[i]) cout << S1[i] << " " << T1[i] << endl;
	else cout << S2[i] << " " << T2[i] << endl;
      }
    }else{
      cout << "Impossible" << endl;
    }
  }
  
  return 0;
}

yukicoder

結果

テストケース

ソースコード