#524135 (C++14) No.1290 Addition and Subtraction Operation

提出ソース

結果

問題	No.1290 Addition and Subtraction Operation
ユーザー	Cinnamoroll
提出日時	2020-08-03 23:13:47
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	123 ms / 2,000 ms
コード長	8,063 bytes
コンパイル時間	2,389 ms
コンパイル使用メモリ	193,048 KB
実行使用メモリ	33,552 KB
最終ジャッジ日時	2023-09-30 02:17:37
合計ジャッジ時間	12,599 ms
ジャッジサーバーID （参考情報）	judge13 / judge11

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	4 ms 6,048 KB
testcase_01	AC	3 ms 6,092 KB
testcase_02	AC	3 ms 5,988 KB
testcase_03	AC	4 ms 6,044 KB
testcase_04	AC	4 ms 5,992 KB
testcase_05	AC	3 ms 6,128 KB
testcase_06	AC	3 ms 6,204 KB
testcase_07	AC	3 ms 5,988 KB
testcase_08	AC	3 ms 5,996 KB
testcase_09	AC	3 ms 6,132 KB
testcase_10	AC	4 ms 6,096 KB
testcase_11	AC	4 ms 6,156 KB
testcase_12	AC	3 ms 6,116 KB
testcase_13	AC	3 ms 6,196 KB
testcase_14	AC	3 ms 6,276 KB
testcase_15	AC	3 ms 6,204 KB
testcase_16	AC	3 ms 6,132 KB
testcase_17	AC	4 ms 5,988 KB
testcase_18	AC	3 ms 5,988 KB
testcase_19	AC	4 ms 6,208 KB
testcase_20	AC	3 ms 6,044 KB
testcase_21	AC	3 ms 6,032 KB
testcase_22	AC	3 ms 6,144 KB
testcase_23	AC	4 ms 6,000 KB
testcase_24	AC	4 ms 6,128 KB
testcase_25	AC	4 ms 6,208 KB
testcase_26	AC	4 ms 5,992 KB
testcase_27	AC	4 ms 6,092 KB
testcase_28	AC	4 ms 5,992 KB
testcase_29	AC	3 ms 6,196 KB
testcase_30	AC	4 ms 6,196 KB
testcase_31	AC	4 ms 5,996 KB
testcase_32	AC	4 ms 5,992 KB
testcase_33	AC	3 ms 6,048 KB
testcase_34	AC	4 ms 6,224 KB
testcase_35	AC	3 ms 6,204 KB
testcase_36	AC	4 ms 6,156 KB
testcase_37	AC	3 ms 6,232 KB
testcase_38	AC	4 ms 5,984 KB
testcase_39	AC	3 ms 5,992 KB
testcase_40	AC	3 ms 5,980 KB
testcase_41	AC	4 ms 6,128 KB
testcase_42	AC	21 ms 8,556 KB
testcase_43	AC	20 ms 8,384 KB
testcase_44	AC	20 ms 8,256 KB
testcase_45	AC	21 ms 8,248 KB
testcase_46	AC	21 ms 8,388 KB
testcase_47	AC	21 ms 8,376 KB
testcase_48	AC	22 ms 8,260 KB
testcase_49	AC	21 ms 8,424 KB
testcase_50	AC	21 ms 8,420 KB
testcase_51	AC	21 ms 8,340 KB
testcase_52	AC	21 ms 8,428 KB
testcase_53	AC	22 ms 8,372 KB
testcase_54	AC	21 ms 8,388 KB
testcase_55	AC	21 ms 8,424 KB
testcase_56	AC	21 ms 8,336 KB
testcase_57	AC	75 ms 31,872 KB
testcase_58	AC	84 ms 30,600 KB
testcase_59	AC	74 ms 32,024 KB
testcase_60	AC	104 ms 27,288 KB
testcase_61	AC	100 ms 28,876 KB
testcase_62	AC	100 ms 28,492 KB
testcase_63	AC	84 ms 30,916 KB
testcase_64	AC	74 ms 33,384 KB
testcase_65	AC	72 ms 32,544 KB
testcase_66	AC	95 ms 30,184 KB
testcase_67	AC	106 ms 27,016 KB
testcase_68	AC	102 ms 28,912 KB
testcase_69	AC	84 ms 30,600 KB
testcase_70	AC	83 ms 31,176 KB
testcase_71	AC	90 ms 30,072 KB
testcase_72	AC	72 ms 32,564 KB
testcase_73	AC	96 ms 29,588 KB
testcase_74	AC	75 ms 33,552 KB
testcase_75	AC	69 ms 32,820 KB
testcase_76	AC	84 ms 30,972 KB
testcase_77	AC	119 ms 28,868 KB
testcase_78	AC	115 ms 29,540 KB
testcase_79	AC	102 ms 31,236 KB
testcase_80	AC	123 ms 27,532 KB
testcase_81	AC	121 ms 27,788 KB
testcase_82	AC	95 ms 30,192 KB
testcase_83	AC	107 ms 27,412 KB
testcase_84	AC	105 ms 28,080 KB
testcase_85	AC	76 ms 32,220 KB
testcase_86	AC	74 ms 33,332 KB

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

ソースコード

raw source code

// warm heart, wagging tail,and a smile just for you!
//                                                                     ███████████
//                                                                   ███╬╬╬╬╬╬╬╬╬╬███
//                                                                ███╬╬╬╬╬████╬╬╬╬╬╬███
//                                            ███████████       ██╬╬╬╬╬████╬╬████╬╬╬╬╬██
//                                      █████████╬╬╬╬╬████████████╬╬╬╬╬██╬╬╬╬╬╬███╬╬╬╬╬██
//                               ████████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬█████████╬╬╬╬╬╬██╬╬╬╬╬╬╬██
//                             ████╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬█████████╬╬╬╬╬╬╬╬╬╬╬██
//                           ███╬╬╬█╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬███╬╬╬╬╬╬╬█████
//                         ███╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬████████╬╬╬╬╬██
//                       ███╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬███╬╬╬╬╬╬╬╬╬███
//                     ███╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬█████╬╬╬╬╬╬╬██
//                 ████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬████╬╬╬╬╬████
//     █████████████╬╬╬╬╬╬╬╬██╬╬╬╬╬████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬█████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬███╬╬╬╬██████
//   ████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬██████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██████╬╬╬╬╬╬╬███████████╬╬╬╬╬╬╬╬██╬╬╬██╬╬╬██
// ███╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬████╬╬╬╬╬╬╬╬╬╬╬█╬╬╬╬╬╬╬██╬╬╬╬╬╬╬╬██
// ██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬╬▓▓▓▓▓▓╬╬╬████╬╬████╬╬╬╬╬╬╬▓▓▓▓▓▓▓▓██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬╬╬╬╬███
// ██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██████▓▓▓▓▓▓▓╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬▓▓▓▓▓▓▓██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██╬╬╬╬█████
// ███╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬███╬╬╬╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬█████╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬████████
//   ███╬╬╬╬╬╬╬╬╬╬╬╬╬█████╬╬╬╬╬╬╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬███╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬██
//       ██████████████  ████╬╬╬╬╬╬███████████████████████████╬╬╬╬╬██╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬╬████
//                         ███████                           █████  ███████████████████  
//
#include "bits/stdc++.h"
using namespace std;
#define INF (1<<30)
#define LINF (1LL<<60)
#define fs first
#define sc second
#define int long long
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define RFOR(i,a,b) for(int i = (b-1);i>=(a);--i)
#define REP(i,n)  FOR(i,0,(n))
#define RREP(i,n) RFOR(i,0,(n))
#define ITR(itr,mp) for(auto itr = (mp).begin(); itr != (mp).end(); ++itr)
#define RITR(itr,mp) for(auto itr = (mp).rbegin(); itr != (mp).rend(); ++itr)
#define range(i,a,b) ((a)<=(i) && (i)<(b))
#define debug(x)  cout << #x << " = " << (x) << endl
#define SP << " " << 
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}
#define MSB(x) (63-__builtin_clzll(x))
#define pcnt(x) (__builtin_popcountll(x))
#define parity(i,j) (i&(1LL<<j))
typedef pair<int,int> P;
typedef tuple<int,int,int> T;
typedef vector<int> vec;
typedef vector<vector<int>> mat;

struct UnionFind {
  private:
  vector<int> child,tree;
  vector<vector<int>> list;

public:
  UnionFind(int v) {
    tree.resize(v);
    list.resize(v);
    REP(i,v) tree[i] = i, list[i].push_back(i);
  }

  int root(int i){
    if(i == tree[i]){
      REP(j,child.size()) tree[child[j]] = i;
      child.clear(); 
      return i;
    }
    else{
      child.push_back(i);
      return i = root(tree[i]);
    }
  }

  int size(int i){
    return list[root(i)].size();
  }

  vector<int> nodes(int no){
    return list[root(no)];
  }

  bool unit(int x, int y){
    x = root(x);
    y = root(y);
    if(x==y) return false;
    if(list[x].size() < list[y].size()) swap(x,y);
    for(int no:list[y]) list[x].emplace_back(no);
    tree[y] = x;
    return true;
  }

  bool isUnit(int x, int y){
    return root(x) == root(y);
  }
};

void ng(){
  cout << "NO" << endl;
  exit(0);
}

const int N = 1e5+10;
vector<int> edge[N],color(N,0);

bool isBip(int no){
  if(!color[no]) color[no] = 1;
  bool flag = true;
  for(auto to:edge[no]){
    if(color[no]==color[to]) flag = false;
    else if(!color[to]){
      color[to] = -color[no];
      if(!isBip(to)) flag = false;
    }
  }
  return flag;
}

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

  int n,m;
  cin >> n >> m;
  
  vector<int> b(n+2);
  REP(i,n) cin >> b[i+1];

  vector<int> d(n+1);
  REP(i,n+1) d[i] = b[i]+b[i+1];

  UnionFind uf(n+1);
  vector<P> op;
  REP(i,m){
    int x,y;
    cin >> x >> y;
    if((y-x)%2) uf.unit(x-1,y);
    else op.emplace_back(x-1,y);
  }

  // 連結成分で座圧
  map<int,int> mp;
  REP(i,n+1) mp[uf.root(i)]++;
  int k = 0; // 連結成分数
  ITR(itr,mp) itr->sc = k++;

  vector<int> id(n+1),sum(k,0); // 座圧後のid, 連結成分内のdのsum
  REP(i,n+1){
    id[i] = mp[uf.root(i)];
    sum[id[i]] += d[i];
  }

  UnionFind uf2(k);
  REP(i,op.size()){
    int x = id[op[i].fs], y = id[op[i].sc];
    uf2.unit(x,y);
    edge[x].emplace_back(y);
    edge[y].emplace_back(x);
  }

  vector<int> list[k];
  REP(i,k) list[uf2.root(i)].emplace_back(i);

  REP(i,k){
    if(uf2.root(i)!=i) continue;
    isBip(i); // 2部グラフを構築
    int v = 0;
    for(int no:list[i]) v += sum[no]*color[no];
    if(v) ng(); // 和が一致する？
  }

  cout << "YES" << endl;

  return 0;
}

yukicoder

結果

テストケース

ソースコード