Submission #204216

#TimeUsernameProblemLanguageResultExecution timeMemory
204216cheetoseDesignated Cities (JOI19_designated_cities)C++11
100 / 100
685 ms57708 KiB
#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]);
	}
}

Compilation message (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 timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...