結果
| 問題 | No.2780 The Bottle Imp |
| コンテスト | |
| ユーザー |
ococonomy1
|
| 提出日時 | 2026-07-25 13:56:16 |
| 言語 | C++17 (gcc 15.2.0 + boost 1.90.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 8,921 bytes |
| 記録 | |
| コンパイル時間 | 2,195 ms |
| コンパイル使用メモリ | 240,960 KB |
| 実行使用メモリ | 36,528 KB |
| 最終ジャッジ日時 | 2026-07-25 13:56:22 |
| 合計ジャッジ時間 | 4,891 ms |
|
ジャッジサーバーID (参考情報) |
judge2_0 / judge3_0 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 36 WA * 4 |
ソースコード
//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;
using pli = pair<ll,int>;
#define MOD 998244353
//#define MOD 1000000007
#define el '\n'
#define El '\n'
#define YESNO(x) ((x) ? "Yes" : "No")
#define YES YESNO(true)
#define NO YESNO(false)
#define EXIT_ANS(x) {cout << (x) << '\n'; return;}
template <typename T> void inline SORT(T &v){sort(v.begin(),v.end()); return;}
template <typename T> void inline REV(T &v){reverse(v.begin(),v.end()); return;}
template <typename T> void inline VEC_UNIQ(T &v){sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end()); return;}
template <typename T> auto inline MAX(T &v){return *max_element(v.begin(),v.end());}
template <typename T> auto inline MIN(T &v){return *min_element(v.begin(),v.end());}
template <typename T> auto inline SUM(vector<T> &v){T ans = 0; for(int i = 0; i < (int)v.size(); i++)ans += v[i]; return ans;}
template <typename T> void inline DEC(T &v){for(int i = 0; i < (int)v.size(); i++)v[i]--; return;}
template <typename T> void inline INC(T &v){for(int i = 0; i < (int)v.size(); i++)v[i]++; return;}
void inline TEST(void){cerr << "TEST" << endl; return;}
template <typename T,typename S> bool inline chmin(T &x,S y){
if(x > (T)y){
x = (T)y;
return true;
}
return false;
}
template <typename T,typename S> bool inline chmax(T &x,S y){
if(x < (T)y){
x = (T)y;
return true;
}
return false;
}
template <typename T = long long> vector<T> inline get_vec(int n){
vector<T> ans(n);
for(int i = 0; i < n; i++)cin >> ans[i];
return ans;
}
template <typename T> void inline print_vec(vector<T> &vec,bool kaigyou = false){
int n = (int)vec.size();
for(int i = 0; i < n; i++){
cout << vec[i];
if(kaigyou || i == n - 1)cout << '\n';
else cout << ' ';
}
if(!n)cout << '\n';
return;
}
template <typename T> void inline debug_vec(vector<T> &vec,bool kaigyou = false){
int n = (int)vec.size();
for(int i = 0; i < n; i++){
cerr << vec[i];
if(kaigyou || i == n - 1)cerr << '\n';
else cerr << ' ';
}
if(!n)cerr << '\n';
return;
}
vector<vector<int>> inline get_graph(int n,int m = -1,bool direct = false,bool decrement = true){
if(m == -1)m = n - 1;
vector<vector<int>> g(n);
while(m--){
int u,v;
cin >> u >> v;
if(decrement)u--,v--;
g[u].push_back(v);
if(!direct)g[v].push_back(u);
}
return g;
}
template <typename T> vector<vector<pair<T,int>>> inline get_weighted_graph(int n,int m = -1,bool direct = false,bool decrement = true){
if(m == -1)m = n - 1;
vector<vector<pair<T,int>>> g(n);
while(m--){
int u,v;
cin >> u >> v;
if(decrement)u--,v--;
ll w; cin >> w;
g[u].push_back(pair(w,v));
if(!direct)g[v].push_back(pair(w,u));
}
return g;
}
// 強連結成分分解(SCC)周りを行うクラス
// 再帰関数周りでグローバル変数使ったりが多いのでクラスで
// ライブラリ書き換えてから未 verify だし、普通に挙動がかなり怪しいので後でちゃんといろいろ検証しておきたい。
// 今は SCC 欲しかったら窃盗する方が安全かも
class ococo_scc {
private:
void dfs(int a) {
checked[a] = true;
for(int i = 0; i < g[a].size(); i++) {
if(!checked[g[a][i]]) {
dfs(g[a][i]);
}
}
junban[nj] = a;
nj++;
}
void dfs2(int a) {
vector<int> ans;
checked[a] = true;
kari.push_back(a);
// cout << a << endl;
for(int i = 0; i < g2[a].size(); i++) {
if(!checked[g2[a][i]]) {
dfs2(g2[a][i]);
}
}
}
public:
vector<vector<int>> g;
vector<vector<int>> g2;
vector<bool> checked;
vector<int> junban;
int nj;
int n;
ococo_scc(int N){
syokica(N);
}
// N個のノードに初期化する O(N)
void syokica(int N) {
n = N;
nj = 0;
g.resize(n);
g2.resize(n);
checked.resize(n);
junban.resize(n);
group.resize(0);
convert.resize(n);
for(int i = 0; i < n; i++) {
g[i].resize(0);
g2[i].resize(0);
checked[i] = false;
}
}
// a→bに向かう有向辺を追加する O(1)
void einsert(int a, int b) {
g[a].push_back(b);
g2[b].push_back(a);
}
// 具体的な有向ループを求めるためのDFSを行う O(V+E)
// 「あるループの成分が全て入ったvector」のvector が返り値
vector<int> kari;
vector<vector<int>> group;
vector<vector<int>> ng;
vector<int> convert;
void build(void) {
for(int i = 0; i < n; i++) {
if(!checked[i]) dfs(i);
}
for(int i = 0; i < n; i++) checked[i] = false;
for(int i = n - 1; i >= 0; i--) {
if(!checked[junban[i]]) {
kari.resize(0);
dfs2(junban[i]);
group.push_back(kari);
}
}
int m = (int)group.size();
for(int i = 0; i < m; i++)sort(group[i].begin(),group[i].end());
auto comp = [](vector<int> &l,vector<int> &r){
return l[0] < r[0];
};
sort(group.begin(),group.end(),comp);
for(int i = 0; i < m; i++){
for(int j = 0; j < (int)group[i].size(); j++){
convert[group[i][j]] = i;
}
}
ng.resize(m);
for(int i = 0; i < n; i++){
for(int j = 0; j < (int)g[i].size(); j++){
int fr = i,to = g[i][j];
if(convert[fr] != convert[to])ng[convert[fr]].push_back(convert[to]);
}
}
for(int i = 0; i < m; i++){
sort(ng[i].begin(),ng[i].end());
ng[i].erase(unique(ng[i].begin(),ng[i].end()),ng[i].end());
}
return;
}
vector<vector<int>> get_newgraph(void){
return ng;
}
vector<int> get_convert(void){
return convert;
}
};
// gはg[a] = {b,c}aからb,cに対して矢印が向いているを意味する配列
// この関数において、順序が自由である部分は辞書順に入るようになっている
//(priority_queの基準を変えればこれは変えることができる)
// gがトポロジカルソート不可能であった場合、サイズ1の-1が要素の配列{-1}を返す
vector<int> ococo_topological_sort(vector<vector<int>> const &g) {
int n = g.size();
vector<int> nyuuzisuu(n, 0);
vector<int> ans(0);
for(int i = 0; i < n; i++)
for(int j = 0; j < g[i].size(); j++) nyuuzisuu[g[i][j]]++;
// que:矢印が自分に来ていない(⇔nyuuzisuu[i]==0である)ノードを辞書順に入れる
priority_queue<int, vector<int>, greater<int>> que;
for(int i = 0; i < n; i++)
if(nyuuzisuu[i] == 0) que.push(i);
while(!que.empty()) {
int kari = que.top();
que.pop();
ans.push_back(kari);
for(int i = 0; i < g[kari].size(); i++) {
nyuuzisuu[g[kari][i]]--;
if(nyuuzisuu[g[kari][i]] == 0) que.push(g[kari][i]);
}
}
for(int i = 0; i < n; i++)
if(nyuuzisuu[i] != 0) {
return {-1};
}
return ans;
}
#define MULTI_TEST_CASE false
void solve(void){
//問題を見たらまず「この問題設定から言えること」をいっぱい言う
//よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く
//複数の解法のアイデアを思いついた時は全部メモしておく
//g++ -D_GLIBCXX_DEBUG -Wall -O2 a.cpp -o o
int n; cin >> n;
ococo_scc scc(n);
for(int i = 0; i < n; i++){
int m; cin >> m;
while(m--){
int a; cin >> a;
a--;
scc.einsert(i,a);
}
}
scc.build();
auto g = scc.get_newgraph();
int m = (int)g.size();
vector<int> conv = scc.get_convert();
vector<int> order = ococo_topological_sort(g);
vector<bool> can_know(m,false);
can_know[conv[0]] = true;
for(int i = 0; i < m; i++){
int idx = order[i];
if(!can_know[idx])EXIT_ANS(NO);
for(int j = 0; j < (int)g[idx].size(); j++){
int to = g[idx][j];
can_know[to] = true;
}
}
cout << YES << El;
return;
}
void calc(void){
return;
}
signed main(void){
cin.tie(nullptr);
ios::sync_with_stdio(false);
calc();
int t = 1;
if(MULTI_TEST_CASE)cin >> t;
while(t--){
solve();
}
return 0;
}
ococonomy1