이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define X first
#define Y second
#define y0 y12
#define y1 y22
#define INF 987654321
#define PI 3.141592653589793238462643383279502884
#define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
#define fdn(i,a,b,c) for(int (i)=(a);(i)>=(b);(i)-=(c))
#define MEM0(a) memset((a),0,sizeof(a));
#define MEM_1(a) memset((a),-1,sizeof(a));
#define ALL(a) a.begin(),a.end()
#define SYNC ios_base::sync_with_stdio(false);cin.tie(0)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef double db;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int, int> Pi;
typedef pair<ll, ll> Pll;
typedef pair<ld, ld> Pd;
typedef vector<int> Vi;
typedef vector<ll> Vll;
typedef vector<ld> Vd;
typedef vector<Pi> VPi;
typedef vector<Pll> VPll;
typedef vector<Pd> VPd;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef tuple<ll, ll, ll> LLL;
typedef vector<iii> Viii;
typedef vector<LLL> VLLL;
typedef complex<double> base;
const int MOD = 1000000007;
ll POW(ll a, ll b, ll MMM = MOD) { ll ret = 1; for (; b; b >>= 1, a = (a*a) % MMM)if (b & 1)ret = (ret*a) % MMM; return ret; }
ll gcd(ll a, ll b) { return b ? gcd(b, a%b) : a; }
ll lcm(ll a, ll b) { if (a == 0 || b == 0)return a + b; return a*(b / gcd(a, b)); }
int dx[] = { 0,1,0,-1,1,1,-1,-1 }, dy[] = { 1,0,-1,0,1,-1,1,-1 };
int ddx[] = { -1,-2,1,-2,2,-1,2,1 }, ddy[] = { -2,-1,-2,1,-1,2,1,2 };
struct A {
int y;
ll c, d;
};
vector<A> v[200001];
VPll vv[200001];
int pt[200001], sz[200001];
ll sum, ans[200001];
ll Q1[200001], q1;
LLL Q2[200001];
ll dfs1(int N, int p = -1)
{
ll res = 0;
for (auto &P : v[N])
{
if (P.y == p)continue;
res += P.d + dfs1(P.y, N);
}
return res;
}
void dfs2(int N, ll now, int p = -1)
{
q1 = max(q1, now);
Q1[N] = now;
for (auto &P : v[N])
{
if (P.y == p)continue;
dfs2(P.y, now + P.c - P.d, N);
}
}
Pll dfs3(int N, int p = -1)
{
VPll V;
for (auto &P : v[N])
{
if (P.y == p)continue;
Pll pp = dfs3(P.y, N);
V.pb(mp(P.c + pp.X, pp.Y));
}
if (V.empty())return mp(0, N);
sort(V.rbegin(), V.rend());
if (V.size()>1)Q2[N] = LLL(Q1[N] + V[0].X + V[1].X, V[0].Y, V[1].Y);
return V[0];
}
ll T1, T2;
priority_queue<LLL> Q;
int ind[200001], parent[200001];
void dfs4(int N, int p = -1)
{
parent[N] = p;
for (auto &P : v[N])
{
if (P.y == p)continue;
dfs4(P.y, N);
}
}
Pll dfs5(int N)
{
for (auto &P : v[N])
{
vv[N].pb(mp(P.y, P.c + dfs5(P.y).Y));
}
sort(ALL(vv[N]), [&](Pll p1, Pll p2) {
return p1.Y>p2.Y;
});
sz[N] = vv[N].size();
fup(i, 0, sz[N] - 1, 1)Q.push(LLL(vv[N][i].Y, N, -i));
if (vv[N].empty())return mp(0, 0);
return vv[N][0];
}
void rem(int T, int R)
{
int NOW = T;
while (NOW != R)
{
int next = parent[NOW];
for (auto &P : v[next])
{
if (P.y == NOW)
{
swap(P, v[next].back());
break;
}
}
v[next].pop_back();
NOW = next;
}
}
int main() {
int n, R;
scanf("%d", &n);
if (n == 2)
{
int a, b, c, d, q;
scanf("%d%d%d%d%d", &a, &b, &c, &d, &q);
while (q--)
{
int x;
scanf("%d", &x);
if (x == 1)printf("%d\n", min(c, d));
else puts("0");
}
return 0;
}
fup(i, 1, n - 1, 1)
{
int x, y;
ll c, d;
scanf("%d%d%lld%lld", &x, &y, &c, &d);
sum += c + d;
v[x].pb({ y,c,d });
v[y].pb({ x,d,c });
}
int leafcnt = 0;
fup(i, 1, n, 1)
{
if (v[i].size()>1)R = i;
else leafcnt++;
}
ll now = dfs1(R);
dfs2(R, now);
int q;
scanf("%d", &q);
ans[1] = q1;
dfs3(R);
R = -1;
LLL mx = LLL(-1, -1, -1);
fup(i, 1, n, 1)
if (Q2[i]>mx)R = i, mx = Q2[i];
T1 = get<1>(mx), T2 = get<2>(mx);
dfs4(R);
fup(i, 1, n, 1)
{
if (i == R)continue;
for (auto &P : v[i])
{
if (P.y == parent[i])
{
swap(P, v[i].back());
break;
}
}
v[i].pop_back();
}
rem(T1, R); rem(T2, R);
fup(i, 1, n, 1)
for (auto &P : v[i])ind[P.y]++;
fup(i, 1, n, 1)if (!ind[i])dfs5(i);
now = get<0>(mx);
ans[2] = now;
int cc = 2;
while (!Q.empty())
{
ll x, N, i;
tie(x, N, i) = Q.top();
i *= -1;
Q.pop();
if (i<pt[N])continue;
cc++, now += x;
if (cc >= 2)ans[cc] = now;
while (pt[N]<sz[N])
{
pt[N]++;
N = get<0>(vv[N][pt[N] - 1]);
}
}
while (q--)
{
int x;
scanf("%d", &x);
if (x >= leafcnt)puts("0");
else printf("%lld\n", sum - ans[x]);
}
}
컴파일 시 표준 에러 (stderr) 메시지
designated_cities.cpp: In function 'int main()':
designated_cities.cpp:134:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &n);
~~~~~^~~~~~~~~~
designated_cities.cpp:138:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d%d%d%d%d", &a, &b, &c, &d, &q);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
designated_cities.cpp:142:9: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &x);
~~~~~^~~~~~~~~~
designated_cities.cpp:152:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d%d%lld%lld", &x, &y, &c, &d);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
designated_cities.cpp:166:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &q);
~~~~~^~~~~~~~~~
designated_cities.cpp:213:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &x);
~~~~~^~~~~~~~~~
designated_cities.cpp:163:17: warning: 'R' may be used uninitialized in this function [-Wmaybe-uninitialized]
ll now = dfs1(R);
^
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |