Submission #853805

# Submission time Handle Problem Language Result Execution time Memory
853805 2023-09-25T09:20:35 Z TimDee Traffickers (RMI18_traffickers) C++17
0 / 100
82 ms 111708 KB
//  Esti <3

//\
     šťastia pre nás :)
//   you're already the best
//             _
//   ^ ^      //
// >(O_O)<___//
//   \ __ __  \
//    \\ \\ \\\\
 
#include <bits/stdc++.h>
using namespace std;
 
//#pragma GCC optimize("O3","unroll-loops")
//#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
 
#pragma GCC optimize("O3")
#pragma GCC target("popcnt")

using ll = long long;
//#define int long long
#define forn(i,n) for(int i=0; i<(n); ++i)
#define pb push_back
#define pi pair<int,int>
#define f first
#define s second 
#define vii(a,n) vector<int> a(n); forn(i,n) cin>>a[i];
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
 
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
const int inf = 1e18;
const int mod = 1e9+7;//998244353;
 
// \
\
:smiling_face_with_3_hearts: :smiling_face_with_3_hearts:  :smiling_face_with_3_hearts:  
 
//vidime sa veľmi skoro, moje slnko

const int N = 30005;

int t[21][21][N];

void add(int i, int j, int k, int x) {
	for(;k<N;k|=k+1) t[i][j][k]+=x;
}
int sum(int i, int j, int r) {
	int q=0;
	for(;r>=0;r=(r&(r+1))-1) q+=t[i][j][r];
	return q;
}

int par[N], top[N], sz[N], d[N];
int ind[N], inv[N];
int nxt=0;
vector<int> adj[N];

void dfs(int u, int p) {
	par[u]=p;
	sz[u]=1;
	for(auto&v:adj[u]) {
		if (v==p) continue;
		d[v]=d[u]+1;
		dfs(v,u);
		sz[u]+=sz[v];
	}
}
void dfs(int u, int p, int t) {
	ind[u]=nxt++;
	inv[ind[u]]=u;
	top[u]=t;
	int mx=-1;
	for(auto&v:adj[u]) {
		if (v==p) continue;
		if (mx==-1) mx=v;
		if (sz[v] > sz[mx]) v=mx;
	}
	if (mx==-1) return;
	dfs(mx,u,t);
	for(auto&v:adj[u]) {
		if (v==p || v==mx) continue;
		dfs(v,u,v);
	}
}

int lca(int u, int v) {
	while (top[u]!=top[v]) {
		if (d[top[u]] < d[top[v]]) swap(u,v);
		u=par[top[u]];
	}
	if (d[u]>d[v]) swap(u,v);
	return u;
}

int f[20];
int s[20];
int fs=0, ss=0;
void qadd(int u, int v, int z) {
	int l = lca(u,v);
	if ((d[l]<=d[u] && d[l]<=d[v])==0) exit(1);
	
	fs=0, ss=0;
	while (u!=l) {
		f[fs++]=u; u=par[u];
	}
	f[fs++]=u;
	while (v!=l) {
		s[ss++]=v; v=par[v];
	}
	for (int i=ss-1; i>=0; --i) f[fs++]=s[i];
	int k=fs;
	return;
	forn(i,k) {
		int u = f[i];
		for (int j=i; j<k; ++j) add(k,j,ind[u],z);
		add(k,20,ind[u],z);
	}
}

ll qry(int u, int t) {
	ll ans=0;
	while (u>=0) {
		forn(i,20) {
			int x = sum(i+1,t%(i+1),ind[u]) - sum(i+1,t%(i+1),ind[top[u]]-1);
			ans+=x;
			int y = sum(i+1,20,ind[u]) - sum(i+1,20,ind[top[u]]-1);
			ans+=1ll*y*(t/(i+1));
		}
		u=par[top[u]];
	}
	return ans;
}
ll query(int u, int v, int t) {
	if (t<0) return 0;
	ll ans = 0;
	int l = lca(u,v);
	ans += qry(u,t);
	ans += qry(v,t);
	ans -= 2ll*qry(l,t);
	forn(i,20) {
		int x = sum(i+1,t%(i+1),ind[l]) - sum(i+1,t%(i+1),ind[l]-1);
		ans+=x;
		int y = sum(i+1,20,ind[l]) - sum(i+1,20,ind[l]-1);
		ans+=1ll*y*(t/(i+1));
	}
	return ans;
}

void solve() {

	int n; cin>>n;
	forn(i,n-1) {
		int u,v; cin>>u>>v; --u, --v;
		adj[u].pb(v);
		adj[v].pb(u);
	}
	dfs(0,-1);
	dfs(0,-1,0);

	int k; cin>>k;
	forn(i,k) {
		int u,v; cin>>u>>v; --u, --v;
		qadd(u,v,1);
	}
	exit(0);
	int q; cin>>q;
	forn(i,q) {
		int t; cin>>t;
		if (t==1) {
			int u,v; cin>>u>>v; --u, --v;
			qadd(u,v,1);
		} else if (t==2) {
			int u,v; cin>>u>>v; --u, --v;
			qadd(u,v,-1);
		} else {
			int u,v,a,b; cin>>u>>v>>a>>b; --u, --v;
			ll ans = query(u,v,b);
			ans -= query(u,v,a-1);
			cout<<ans<<'\n';
		}
	}

}

int32_t main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int t=1;
    //cin>>t;
    while (t--) solve();
    return 0;
}

Compilation message

traffickers.cpp:3:1: warning: multi-line comment [-Wcomment]
    3 | //\
      | ^
traffickers.cpp:9:1: warning: multi-line comment [-Wcomment]
    9 | //   \ __ __  \
      | ^
traffickers.cpp:36:1: warning: multi-line comment [-Wcomment]
   36 | // \
      | ^
traffickers.cpp:33:17: warning: overflow in conversion from 'double' to 'int' changes value from '1.0e+18' to '2147483647' [-Woverflow]
   33 | const int inf = 1e18;
      |                 ^~~~
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2648 KB Output isn't correct
2 Incorrect 1 ms 2648 KB Output isn't correct
3 Runtime error 74 ms 108368 KB Execution killed with signal 11
# Verdict Execution time Memory Grader output
1 Runtime error 75 ms 109136 KB Execution killed with signal 11
2 Runtime error 75 ms 109060 KB Execution killed with signal 11
3 Runtime error 74 ms 109364 KB Execution killed with signal 11
4 Runtime error 74 ms 109232 KB Execution killed with signal 11
5 Runtime error 78 ms 109176 KB Execution killed with signal 11
6 Runtime error 77 ms 109220 KB Execution killed with signal 11
7 Runtime error 75 ms 109148 KB Execution killed with signal 11
8 Runtime error 74 ms 109392 KB Execution killed with signal 11
9 Incorrect 4 ms 3160 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Runtime error 79 ms 110888 KB Execution killed with signal 11
2 Runtime error 82 ms 111708 KB Execution killed with signal 11
3 Runtime error 79 ms 111184 KB Execution killed with signal 11
4 Runtime error 80 ms 110416 KB Execution killed with signal 11
5 Runtime error 79 ms 110224 KB Execution killed with signal 11
6 Runtime error 81 ms 111440 KB Execution killed with signal 11
7 Incorrect 18 ms 4696 KB Output isn't correct
8 Incorrect 19 ms 4184 KB Output isn't correct