#include <bits/stdc++.h> #include <chrono> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; //using namespace chrono; using namespace __gnu_pbds; #pragma GCC optimize("Ofast") //#pragma GCC optimize("Ofast,unroll-loops") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,fma,tune=native") //#pragma GCC target("sse,sse2,sse3,mmx,abm,tune=native") // for szkopul and sio only typedef long long lld; typedef double lf; typedef long double llf; typedef pair<int,int> pii; typedef pair<lld,lld> pll; typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> oset; #define For(i,s,a) for(lld i = (lld)s; i < (lld)a; ++i) #define rpt(s, it) for(auto it = s.begin(); it != s.end(); ++it) #define brpt(s, it) for(auto it = s.rend(); it != s.rbegin(); --it) #define pb push_back #define eb emplace_back #define ff first #define dd second #define mp make_pair #define all(x) (x).begin(), (x).end() #define make_unique(x) (x).erase( unique(all(x)), (x).end()) #define popcnt(x) __builtin_popcount(x) #define sz size() #define time_since duration_cast< milliseconds >(system_clock::now().time_since_epoch()) template<typename Ta, typename Tb> ostream & operator <<(ostream & os, pair<Ta, Tb> x){ return os << x.ff << " " << x.dd; } const int N = 1e4 + 1; short int edges[N][N]; bitset<N>contents[N]; short int sizes[N]; void solve(void) { int n; scanf("%d", &n); For(i, 1, n + 1) { scanf("%hd", &sizes[i]); For(j, 0, sizes[i]) scanf("%hd", &edges[i][j]); sort(edges[i], edges[i] + sizes[i]); } For(k, 1, n + 1) { For(i, 1, k + 1) contents[i] = 0; For(i, 1, k + 1) if(edges[i][sizes[i] - 1] <= k && sizes[i] == 1) { For(j, 0, sizes[i]) contents[i][edges[i][j]] = 1; contents[i][i] = 1; } For(iter, 0, k) For(i, 1, k + 1) { if(edges[i][sizes[i] - 1] > k) continue; contents[i] = contents[edges[i][0]]; For(j, 1, sizes[i]) contents[i] &= contents[edges[i][j]]; if(sizes[i] == 1) contents[i][edges[i][0]] = 1; contents[i][i] = 1; } int wyn = 0; For(i, 1, k + 1) { /*cout << k << " " << i << " " << contents[i].count() << " " << contents[i][i] << endl; For(j, 0, n + 1) cout << contents[i][j] << " "; puts("");*/ wyn += contents[i].count() - (contents[i][i] ? 1 : 0); } printf("%d ", wyn); /*puts(""); printf("%lld\n", k); For(i, 1, k + 1) For(j, 1, k + 1) if(contents[i][j] && i != j) cout << i << " " << j << endl; puts("\n\n");*/ } } int32_t main(void){ int t = 1; //scanf("%d", &t); while(t--) solve(); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | #include <bits/stdc++.h> #include <chrono> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; //using namespace chrono; using namespace __gnu_pbds; #pragma GCC optimize("Ofast") //#pragma GCC optimize("Ofast,unroll-loops") //#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,fma,tune=native") //#pragma GCC target("sse,sse2,sse3,mmx,abm,tune=native") // for szkopul and sio only typedef long long lld; typedef double lf; typedef long double llf; typedef pair<int,int> pii; typedef pair<lld,lld> pll; typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> oset; #define For(i,s,a) for(lld i = (lld)s; i < (lld)a; ++i) #define rpt(s, it) for(auto it = s.begin(); it != s.end(); ++it) #define brpt(s, it) for(auto it = s.rend(); it != s.rbegin(); --it) #define pb push_back #define eb emplace_back #define ff first #define dd second #define mp make_pair #define all(x) (x).begin(), (x).end() #define make_unique(x) (x).erase( unique(all(x)), (x).end()) #define popcnt(x) __builtin_popcount(x) #define sz size() #define time_since duration_cast< milliseconds >(system_clock::now().time_since_epoch()) template<typename Ta, typename Tb> ostream & operator <<(ostream & os, pair<Ta, Tb> x){ return os << x.ff << " " << x.dd; } const int N = 1e4 + 1; short int edges[N][N]; bitset<N>contents[N]; short int sizes[N]; void solve(void) { int n; scanf("%d", &n); For(i, 1, n + 1) { scanf("%hd", &sizes[i]); For(j, 0, sizes[i]) scanf("%hd", &edges[i][j]); sort(edges[i], edges[i] + sizes[i]); } For(k, 1, n + 1) { For(i, 1, k + 1) contents[i] = 0; For(i, 1, k + 1) if(edges[i][sizes[i] - 1] <= k && sizes[i] == 1) { For(j, 0, sizes[i]) contents[i][edges[i][j]] = 1; contents[i][i] = 1; } For(iter, 0, k) For(i, 1, k + 1) { if(edges[i][sizes[i] - 1] > k) continue; contents[i] = contents[edges[i][0]]; For(j, 1, sizes[i]) contents[i] &= contents[edges[i][j]]; if(sizes[i] == 1) contents[i][edges[i][0]] = 1; contents[i][i] = 1; } int wyn = 0; For(i, 1, k + 1) { /*cout << k << " " << i << " " << contents[i].count() << " " << contents[i][i] << endl; For(j, 0, n + 1) cout << contents[i][j] << " "; puts("");*/ wyn += contents[i].count() - (contents[i][i] ? 1 : 0); } printf("%d ", wyn); /*puts(""); printf("%lld\n", k); For(i, 1, k + 1) For(j, 1, k + 1) if(contents[i][j] && i != j) cout << i << " " << j << endl; puts("\n\n");*/ } } int32_t main(void){ int t = 1; //scanf("%d", &t); while(t--) solve(); } |