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

提出ソース

結果

問題	No.1290 Addition and Subtraction Operation
ユーザー	Cinnamoroll
提出日時	2020-08-03 22:24:52
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	140 ms / 2,000 ms
コード長	8,392 bytes
コンパイル時間	2,538 ms
コンパイル使用メモリ	200,472 KB
実行使用メモリ	41,472 KB
最終ジャッジ日時	2024-07-22 20:12:54
合計ジャッジ時間	8,805 ms
ジャッジサーバーID （参考情報）	judge2 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	4 ms 6,400 KB
testcase_01	AC	4 ms 6,272 KB
testcase_02	AC	5 ms 6,400 KB
testcase_03	AC	4 ms 6,400 KB
testcase_04	AC	4 ms 6,400 KB
testcase_05	AC	4 ms 6,400 KB
testcase_06	AC	4 ms 6,400 KB
testcase_07	AC	4 ms 6,400 KB
testcase_08	AC	4 ms 6,400 KB
testcase_09	AC	4 ms 6,400 KB
testcase_10	AC	4 ms 6,400 KB
testcase_11	AC	4 ms 6,400 KB
testcase_12	AC	4 ms 6,400 KB
testcase_13	AC	4 ms 6,272 KB
testcase_14	AC	4 ms 6,400 KB
testcase_15	AC	4 ms 6,400 KB
testcase_16	AC	4 ms 6,400 KB
testcase_17	AC	4 ms 6,272 KB
testcase_18	AC	4 ms 6,400 KB
testcase_19	AC	4 ms 6,400 KB
testcase_20	AC	4 ms 6,400 KB
testcase_21	AC	4 ms 6,400 KB
testcase_22	AC	4 ms 6,528 KB
testcase_23	AC	4 ms 6,400 KB
testcase_24	AC	3 ms 6,400 KB
testcase_25	AC	4 ms 6,400 KB
testcase_26	AC	4 ms 6,272 KB
testcase_27	AC	4 ms 6,400 KB
testcase_28	AC	4 ms 6,272 KB
testcase_29	AC	4 ms 6,400 KB
testcase_30	AC	4 ms 6,400 KB
testcase_31	AC	4 ms 6,400 KB
testcase_32	AC	4 ms 6,400 KB
testcase_33	AC	4 ms 6,400 KB
testcase_34	AC	4 ms 6,400 KB
testcase_35	AC	4 ms 6,400 KB
testcase_36	AC	3 ms 6,400 KB
testcase_37	AC	4 ms 6,400 KB
testcase_38	AC	4 ms 6,400 KB
testcase_39	AC	4 ms 6,400 KB
testcase_40	AC	4 ms 6,400 KB
testcase_41	AC	4 ms 6,272 KB
testcase_42	AC	20 ms 8,392 KB
testcase_43	AC	21 ms 8,512 KB
testcase_44	AC	20 ms 8,516 KB
testcase_45	AC	20 ms 8,520 KB
testcase_46	AC	21 ms 8,488 KB
testcase_47	AC	20 ms 8,644 KB
testcase_48	AC	20 ms 8,520 KB
testcase_49	AC	21 ms 8,516 KB
testcase_50	AC	20 ms 8,516 KB
testcase_51	AC	20 ms 8,516 KB
testcase_52	AC	21 ms 8,516 KB
testcase_53	AC	21 ms 8,516 KB
testcase_54	AC	21 ms 8,520 KB
testcase_55	AC	21 ms 8,512 KB
testcase_56	AC	21 ms 8,512 KB
testcase_57	AC	97 ms 38,144 KB
testcase_58	AC	107 ms 35,788 KB
testcase_59	AC	96 ms 38,400 KB
testcase_60	AC	109 ms 29,432 KB
testcase_61	AC	109 ms 32,228 KB
testcase_62	AC	109 ms 31,336 KB
testcase_63	AC	105 ms 36,048 KB
testcase_64	AC	95 ms 41,216 KB
testcase_65	AC	89 ms 39,808 KB
testcase_66	AC	109 ms 34,364 KB
testcase_67	AC	109 ms 29,016 KB
testcase_68	AC	111 ms 31,960 KB
testcase_69	AC	105 ms 35,908 KB
testcase_70	AC	103 ms 36,680 KB
testcase_71	AC	110 ms 34,636 KB
testcase_72	AC	94 ms 39,552 KB
testcase_73	AC	111 ms 33,356 KB
testcase_74	AC	101 ms 41,472 KB
testcase_75	AC	89 ms 40,320 KB
testcase_76	AC	107 ms 36,420 KB
testcase_77	AC	135 ms 31,836 KB
testcase_78	AC	140 ms 32,964 KB
testcase_79	AC	130 ms 36,992 KB
testcase_80	AC	132 ms 29,916 KB
testcase_81	AC	129 ms 30,064 KB
testcase_82	AC	117 ms 34,508 KB
testcase_83	AC	117 ms 29,908 KB
testcase_84	AC	115 ms 30,808 KB
testcase_85	AC	98 ms 38,784 KB
testcase_86	AC	95 ms 41,088 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];
  }

  vector<int> flag(k,0); // 自己ループを含むか
  UnionFind uf2(k);
  REP(i,op.size()){
    int x = id[op[i].fs], y = id[op[i].sc];
    if(x==y) flag[x]++;
    else{
      uf2.unit(x,y);
      edge[x].emplace_back(y);
      edge[y].emplace_back(x);
    }
  }

  map<int,int> sum2;
  vector<int> flag2(k,0), list[k];
  REP(i,k){
    sum2[uf2.root(i)] += sum[i];
    flag2[uf2.root(i)] += flag[i];
    list[uf2.root(i)].emplace_back(i);
  }

  REP(i,k){
    if(uf2.root(i)!=i) continue;
    if(sum2[i]%2) ng(); // 総和を奇数にはできない
    if(flag2[i]) continue; // 自己ループ
    if(!isBip(i)) continue; // 奇閉路ある？
    int v = 0;
    for(int no:list[i]) v += sum[no]*color[no];
    if(v) ng(); // 和が一致する？
  }

  cout << "YES" << endl;

  return 0;
}

yukicoder

結果

テストケース

ソースコード