Submission #1370519
Source Code Expand
#include <bits/stdc++.h> using namespace std; #define FOR(i,a,b) for(int i=(a);i<(b);i++) #define REP(i,n) FOR(i,0,n) #define ALL(v) (v).begin(),(v).end() #define fi first #define se second template<typename A, typename B> inline bool chmax(A &a, B b) { if (a<b) { a=b; return 1; } return 0; } template<typename A, typename B> inline bool chmin(A &a, B b) { if (a>b) { a=b; return 1; } return 0; } typedef long long ll; typedef pair<int, int> pii; typedef pair<ll, ll> pll; typedef pair<int, pll> P; const ll INF = 1ll<<30; const ll MOD = 1000000007; const double EPS = 1e-10; int N; int a[112345]; vector<int> g[112345]; vector<pll> buf[112345]; int main() { cin >> N; REP(i, N) scanf("%d", a + i); REP(i, N - 1) { int a, b; scanf("%d %d", &a, &b); a--; b--; g[a].push_back(b); g[b].push_back(a); } queue<pii> que; // 現在の頂点, どの頂点からきたか REP(i, N) if (g[i].size() == 1) { int v = g[i][0]; que.push(pii(v, i)); buf[v].push_back(pll(a[i], i)); } bool ans = true; while (!que.empty()) { pii now = que.front(); que.pop(); int u = now.fi, bef = now.se; ll sum = 0; REP(i, buf[u].size()) sum += buf[u][i].fi; if (buf[u].size() + 1 == g[u].size()) { ll to = 2 * a[u] - sum; set<int> ss; REP(i, buf[u].size()) ss.insert(buf[u][i].se); REP(i, g[u].size()) { int v = g[u][i]; if (v == bef || ss.find(v) != ss.end()) continue; que.push(pii(v, u)); buf[v].push_back(pll(to, u)); } } else if (buf[u].size() == g[u].size()) { if (sum != 2 * a[u]) ans = false; } } puts(ans ? "YES" : "NO"); return 0; }
Submission Info
Submission Time | |
---|---|
Task | C - Cleaning |
User | tkmst201 |
Language | C++14 (GCC 5.4.1) |
Score | 0 |
Code Size | 1704 Byte |
Status | WA |
Exec Time | 78 ms |
Memory | 12160 KB |
Compile Error
./Main.cpp: In function ‘int main()’: ./Main.cpp:30:30: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] REP(i, N) scanf("%d", a + i); ^ ./Main.cpp:33:25: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result] scanf("%d %d", &a, &b); ^
Judge Result
Set Name | Sample | All | ||||||
---|---|---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 0 / 700 | ||||||
Status |
|
|
Set Name | Test Cases |
---|---|
Sample | sample1.txt, sample2.txt, sample3.txt |
All | sample1.txt, sample2.txt, sample3.txt, in1.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in2.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in3.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in4.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in5.txt, in6.txt, in7.txt, in8.txt, in9.txt, sample1.txt, sample2.txt, sample3.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
in1.txt | WA | 72 ms | 12160 KB |
in10.txt | WA | 71 ms | 12160 KB |
in11.txt | AC | 66 ms | 12160 KB |
in12.txt | AC | 62 ms | 12160 KB |
in13.txt | AC | 56 ms | 12160 KB |
in14.txt | AC | 59 ms | 12160 KB |
in15.txt | AC | 4 ms | 5504 KB |
in16.txt | AC | 4 ms | 5504 KB |
in17.txt | WA | 4 ms | 5504 KB |
in18.txt | AC | 4 ms | 5504 KB |
in19.txt | AC | 72 ms | 12160 KB |
in2.txt | WA | 71 ms | 12160 KB |
in20.txt | AC | 71 ms | 12160 KB |
in21.txt | AC | 78 ms | 12160 KB |
in22.txt | AC | 73 ms | 12160 KB |
in23.txt | AC | 70 ms | 12160 KB |
in24.txt | AC | 72 ms | 12160 KB |
in25.txt | WA | 4 ms | 5504 KB |
in26.txt | AC | 72 ms | 12160 KB |
in27.txt | AC | 72 ms | 12160 KB |
in28.txt | AC | 4 ms | 5504 KB |
in29.txt | AC | 68 ms | 12160 KB |
in3.txt | WA | 70 ms | 12160 KB |
in30.txt | WA | 4 ms | 5504 KB |
in31.txt | AC | 70 ms | 12160 KB |
in32.txt | AC | 70 ms | 12160 KB |
in33.txt | WA | 70 ms | 12160 KB |
in34.txt | WA | 70 ms | 12160 KB |
in35.txt | AC | 70 ms | 12160 KB |
in36.txt | AC | 73 ms | 12160 KB |
in37.txt | AC | 4 ms | 5632 KB |
in38.txt | AC | 71 ms | 12160 KB |
in39.txt | AC | 71 ms | 12160 KB |
in4.txt | WA | 70 ms | 12160 KB |
in40.txt | AC | 70 ms | 12160 KB |
in41.txt | AC | 72 ms | 12160 KB |
in42.txt | AC | 72 ms | 12160 KB |
in43.txt | AC | 70 ms | 12160 KB |
in44.txt | AC | 70 ms | 12160 KB |
in45.txt | AC | 70 ms | 12160 KB |
in5.txt | WA | 70 ms | 12160 KB |
in6.txt | WA | 70 ms | 12160 KB |
in7.txt | WA | 62 ms | 11392 KB |
in8.txt | WA | 18 ms | 7040 KB |
in9.txt | WA | 72 ms | 12160 KB |
sample1.txt | AC | 4 ms | 5504 KB |
sample2.txt | AC | 3 ms | 5504 KB |
sample3.txt | AC | 4 ms | 5504 KB |