| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 858219 | 2023-10-07T16:05:52 Z | efedmrlr | 조이터에서 친구를 만드는건 재밌어 (JOI20_joitter2) | C++17 | 13 ms | 30068 KB |
#include <bits/stdc++.h>
using namespace std;
#define n_l '\n'
#define dbg(...) cout << "[" << #__VA_ARGS__ << "]: "; cout << to_string(__VA_ARGS__) << endl
template <typename T, size_t N> int SIZE(const T (&t)[N]){ return N; } template<typename T> int SIZE(const T &t){ return t.size(); } string to_string(const string s, int x1=0, int x2=1e9){ return '"' + ((x1 < s.size()) ? s.substr(x1, x2-x1+1) : "") + '"'; } string to_string(const char* s) { return to_string((string) s); } string to_string(const bool b) { return (b ? "true" : "false"); } string to_string(const char c){ return string({c}); } template<size_t N> string to_string(const bitset<N> &b, int x1=0, int x2=1e9){ string t = ""; for(int __iii__ = min(x1,SIZE(b)), __jjj__ = min(x2, SIZE(b)-1); __iii__ <= __jjj__; ++__iii__){ t += b[__iii__] + '0'; } return '"' + t + '"'; } template <typename A, typename... C> string to_string(const A (&v), int x1=0, int x2=1e9, C... coords); int l_v_l_v_l = 0, t_a_b_s = 0; template <typename A, typename B> string to_string(const pair<A, B> &p) { l_v_l_v_l++; string res = "(" + to_string(p.first) + ", " + to_string(p.second) + ")"; l_v_l_v_l--; return res; } template <typename A, typename... C> string to_string(const A (&v), int x1, int x2, C... coords) { int rnk = rank<A>::value; string tab(t_a_b_s, ' '); string res = ""; bool first = true; if(l_v_l_v_l == 0) res += n_l; res += tab + "["; x1 = min(x1, SIZE(v)), x2 = min(x2, SIZE(v)); auto l = begin(v); advance(l, x1); auto r = l; advance(r, (x2-x1) + (x2 < SIZE(v))); for (auto e = l; e != r; e = next(e)) { if (!first) { res += ", "; } first = false; l_v_l_v_l++; if(e != l){ if(rnk > 1) { res += n_l; t_a_b_s = l_v_l_v_l; }; } else{ t_a_b_s = 0; } res += to_string(*e, coords...); l_v_l_v_l--; } res += "]"; if(l_v_l_v_l == 0) res += n_l; return res; } void dbgm(){;} template<typename Heads, typename... Tails> void dbgm(Heads H, Tails... T){ cout << to_string(H) << " | "; dbgm(T...); }
#define dbgm(...) cout << "[" << #__VA_ARGS__ << "]: "; dbgm(__VA_ARGS__); cout << endl
#define int long long int
#define MP make_pair
#define pb push_back
#define REP(i,n) for(int (i) = 0; (i) < (n); (i)++)
void fastio() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
}
const double EPS = 0.00001;
const int INF = 1e9+500;
const int N = 1e5+5;
const int M = 3e5+5;
const int ALPH = 26;
const int LGN = 25;
const int MOD = 1e9+7;
int n,m,q;
int p[N];
int sz[N];
set<int> st[N], outst[N], childs[N];
int ans = 0;
queue<array<int,2> > upds;
void make_set(int x) {
p[x] = x;
st[x].clear();
st[x].insert(x);
sz[x] = 1;
outst[x].clear();
childs[x].clear();
childs[x].insert(x);
}
int find_set(int x) {
if(p[x] == x) return x;
return find_set(p[x]);
}
void insert_weak(int a, int b) {
if(st[b].count(a) ) {
assert(outst[a].count(b));
return;
}
st[b].insert(a);
outst[a].insert(b);
if(st[a].count(b)) {
assert(outst[b].count(a));
upds.push({a,b});
}
}
int dsu_size(int x) {
return st[x].size() + outst[x].size() + childs[x].size();
}
void union_set(int x, int y) {
x = find_set(x);
y = find_set(y);
if(x==y) return;
if(dsu_size(x) < dsu_size(y)) swap(x,y);
ans -= (childs[x].size() - sz[x]) * (sz[x]) + (childs[y].size() - sz[y]) * (sz[y]);
ans -= sz[x] * (sz[x] - 1) + sz[y] * (sz[y] - 1);
p[y] = x;
sz[x] += sz[y];
st[x].erase(y), outst[x].erase(y);
st[y].erase(x), outst[y].erase(x);
for(auto el : outst[y]) {
st[el].erase(y);
insert_weak(x, el);
}
for(auto el : st[y]) {
outst[el].erase(y);
insert_weak(el, x);
}
for(auto el : childs[y]) {
childs[x].insert(el);
}
ans += sz[x] * (sz[x] - 1);
ans += sz[x] * (childs[x].size() - sz[x]);
}
inline void solve() {
cin>>n>>m;
for(int i = 1; i<=n; i++) {
make_set(i);
}
for(int i = 0; i<m; i++) {
int a,b;
cin>>a>>b;
b = find_set(b);
if(find_set(a) != b && !(childs[b].count(a))) {
ans += sz[b];
childs[b].insert(a);
a = find_set(a);
insert_weak(a, b);
}
while(upds.size()) {
array<int,2> tmp = upds.front();
upds.pop();
union_set(tmp[0], tmp[1]);
}
cout<<ans<<"\n";
}
}
signed main() {
//freopen("milkvisits.in", "r", stdin);
//freopen("milkvisits.out", "w", stdout);
fastio();
int test = 1;
//cin>>test;
while(test--) {
solve();
}
}
Compilation message
| # | 결과 | 실행 시간 | 메모리 | Grader output |
|---|---|---|---|---|
| 1 | Correct | 3 ms | 14996 KB | Output is correct |
| 2 | Correct | 3 ms | 14936 KB | Output is correct |
| 3 | Correct | 3 ms | 14940 KB | Output is correct |
| 4 | Correct | 3 ms | 14940 KB | Output is correct |
| 5 | Correct | 3 ms | 14940 KB | Output is correct |
| 6 | Correct | 4 ms | 14772 KB | Output is correct |
| 7 | Runtime error | 13 ms | 30068 KB | Execution killed with signal 6 |
| 8 | Halted | 0 ms | 0 KB | - |
| # | 결과 | 실행 시간 | 메모리 | Grader output |
|---|---|---|---|---|
| 1 | Correct | 3 ms | 14996 KB | Output is correct |
| 2 | Correct | 3 ms | 14936 KB | Output is correct |
| 3 | Correct | 3 ms | 14940 KB | Output is correct |
| 4 | Correct | 3 ms | 14940 KB | Output is correct |
| 5 | Correct | 3 ms | 14940 KB | Output is correct |
| 6 | Correct | 4 ms | 14772 KB | Output is correct |
| 7 | Runtime error | 13 ms | 30068 KB | Execution killed with signal 6 |
| 8 | Halted | 0 ms | 0 KB | - |
| # | 결과 | 실행 시간 | 메모리 | Grader output |
|---|---|---|---|---|
| 1 | Correct | 3 ms | 14996 KB | Output is correct |
| 2 | Correct | 3 ms | 14936 KB | Output is correct |
| 3 | Correct | 3 ms | 14940 KB | Output is correct |
| 4 | Correct | 3 ms | 14940 KB | Output is correct |
| 5 | Correct | 3 ms | 14940 KB | Output is correct |
| 6 | Correct | 4 ms | 14772 KB | Output is correct |
| 7 | Runtime error | 13 ms | 30068 KB | Execution killed with signal 6 |
| 8 | Halted | 0 ms | 0 KB | - |