oj.uz

답안 #347377

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
347377 2021-01-12T16:28:52 Z cheetose 도로 개발 (JOI15_road_development) C++17
25 / 100
 842 ms 26348 KB 
#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 1987654321
#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 COMPRESS(a) sort(ALL(a));a.resize(unique(ALL(a))-a.begin())
#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<db, db> Pd;
typedef vector<int> Vi;
typedef vector<ll> Vll;
typedef vector<double> 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; }
int dx[] = { 0,1,0,-1,1,1,-1,-1 }, dy[] = { 1,0,-1,0,1,-1,1,-1 };
int ddx[]={2,2,-2,-2,1,1,-1,-1},ddy[]={1,-1,1,-1,2,-2,2,-2};

Vi v[100001];
int p[100001];
int find(int x){
	if(p[x]==x)return x;
	return p[x]=find(p[x]);
}
void merge(int x,int y){
	x=find(x),y=find(y);
	if(x!=y)p[x]=y;
}
int tree[262144],lazy[262144],hld[100001],cnt[100001],num[100001],idx[100001],parent[100001],depth[100001],cc;
void init(int node,int S, int E){
	if(S==E){
		tree[node]=1;
		return;
	}
	init(node*2,S,(S+E)/2);
	init(node*2+1,(S+E)/2+1,E);
	tree[node]=tree[node*2]+tree[node*2+1];
}
void propagation(int node,int S,int E){
	if(lazy[node]){
		tree[node]=0;
		if(S!=E){
			lazy[node*2]=1;
			lazy[node*2+1]=1;
		}
		lazy[node]=0;
	}
}
void upd(int node, int S, int E, int i, int j)
{
	propagation(node, S, E);
	if (i > E || j < S) return;
	if (j >= E && i <= S)
	{
		lazy[node] = 1;
		propagation(node, S, E);
		return;
	}
	upd(2 * node, S, (S + E) / 2, i, j);
	upd(2 * node + 1, (S + E) / 2 + 1, E, i, j);
	tree[node] = tree[2 * node] + tree[2 * node + 1];
}

int find(int node, int S, int E, int i, int j)
{
	propagation(node, S, E);
	if (i > E || j < S)return 0;
	if (i <= S && j >= E)return tree[node];
	return find(node * 2, S, (S + E) / 2, i, j)+ find(node * 2 + 1, (S + E) / 2 + 1, E, i, j);
}
void dfs(int N,int p,int d)
{
	parent[N]=p;
	depth[N]=d;
	cnt[N]=1;
	for(int next:v[N])
	{
		if(next==p)continue;
		dfs(next,N,d+1);
		cnt[N]+=cnt[next];
	}
}
void HLD(int N,int p)
{
	int c=-1;
	num[N]=++cc;
	idx[num[N]]=N;
	for(int next:v[N])
	{
		if(next==p)continue;
		if(c==-1 || cnt[next]>cnt[c])c=next;
	}
	if(c!=-1)
	{
		hld[c]=hld[N];
		HLD(c,N);
	}
	for(int next:v[N])
	{
		if(next==p || next==c)continue;
		hld[next]=next;
		HLD(next,N);
	}
}
int n,q;
void upd(int x,int y){
	while(hld[x]!=hld[y])
	{
		if(depth[hld[x]]>depth[hld[y]])
		{
			upd(1,1,n,num[hld[x]],num[x]);
			x=parent[hld[x]];
		}
		else
		{
			upd(1,1,n,num[hld[y]],num[y]);
			y=parent[hld[y]];
		}
	}
	if(depth[x]>depth[y])swap(x,y);
	upd(1,1,n,num[x]+1,num[y]);
}
int query(int x,int y)
{
	int res=0;
	while(hld[x]!=hld[y])
	{
		if(depth[hld[x]]>depth[hld[y]])
		{
			res+=find(1,1,n,num[hld[x]],num[x]);
			x=parent[hld[x]];
		}
		else
		{
			res+=find(1,1,n,num[hld[y]],num[y]);
			y=parent[hld[y]];
		}
	}
	if(depth[x]>depth[y])swap(x,y);
	res+=find(1,1,n,num[x]+1,num[y]);
	return res;
}
iii Q[300000];
int main() {
	scanf("%d%d",&n,&q);
	iota(p,p+n+1,0);
	fup(i,0,q-1,1){
		int t,x,y;
		scanf("%d%d%d",&t,&x,&y);
		Q[i]=iii(t,x,y);
		if(find(x)!=find(y)){
			v[x].pb(y);
			v[y].pb(x);
		}
		merge(x,y);
	}
	fup(i,1,n,1){
		if(!num[i]){
			dfs(i,-1,0);
			hld[i]=i;
			HLD(i,-1);
		}
	}
	init(1,1,n);
	iota(p,p+n+1,0);
	fup(i,0,q-1,1){
		auto [t,x,y]=Q[i];
		if(t==1){
			if(find(x)==find(y)){
				upd(x,y);
			}
			merge(x,y);
		}else{
			if(find(x)!=find(y)){
				puts("-1");
				continue;
			}
			printf("%d\n",query(x,y));
		}
	}
}

Compilation message

road_development.cpp: In function 'int main()':
road_development.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
road_development.cpp:171:2: note: in expansion of macro 'fup'
  171 |  fup(i,0,q-1,1){
      |  ^~~
road_development.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
road_development.cpp:181:2: note: in expansion of macro 'fup'
  181 |  fup(i,1,n,1){
      |  ^~~
road_development.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
road_development.cpp:190:2: note: in expansion of macro 'fup'
  190 |  fup(i,0,q-1,1){
      |  ^~~
road_development.cpp:169:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  169 |  scanf("%d%d",&n,&q);
      |  ~~~~~^~~~~~~~~~~~~~
road_development.cpp:173:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  173 |   scanf("%d%d%d",&t,&x,&y);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 2816 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
25.0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Correct 387 ms 18572 KB Output is correct
2 Correct 842 ms 18540 KB Output is correct
3 Correct 352 ms 24072 KB Output is correct
4 Correct 366 ms 24172 KB Output is correct
5 Correct 435 ms 22380 KB Output is correct
6 Correct 369 ms 22380 KB Output is correct
7 Correct 679 ms 18668 KB Output is correct
8 Correct 337 ms 22124 KB Output is correct
9 Correct 652 ms 18156 KB Output is correct
10 Correct 359 ms 20992 KB Output is correct
11 Correct 308 ms 23404 KB Output is correct
12 Correct 317 ms 26012 KB Output is correct
13 Correct 275 ms 26348 KB Output is correct
14 Correct 328 ms 22764 KB Output is correct

Subtask #3
0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 383 ms 18532 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Correct 258 ms 15716 KB Output is correct
2 Correct 532 ms 15936 KB Output is correct
3 Correct 320 ms 18284 KB Output is correct
4 Correct 301 ms 18924 KB Output is correct
5 Correct 834 ms 18116 KB Output is correct
6 Correct 470 ms 21208 KB Output is correct
7 Incorrect 83 ms 13164 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #5
0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 2816 KB Output isn't correct
2 Halted 0 ms 0 KB -