#include <bits/stdc++.h>
#define jizz ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define pb push_back
#define ET cout << "\n"
#define MEM(i,j) memset(i,j,sizeof i)
#define F first
#define S second
#define MP make_pair
#define ALL(v) v.begin(),v.end()
#define DB(a,s,e) {for(int i=s;i<e;++i) cout << a[i] << " ";ET;}
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
const ll MOD=1e9;
ll dx[4]={1,-1,0,0},dy[4]={0,0,1,-1},ans,boss[100005];
pll val[100005],dots[100005];
vector<ll> v,hav[100005];
unordered_map<ll,ll> mpp;
vector<ll> G[100005];
bitset<100005> vis;
ll finds(ll x)
{
if(x==boss[x]) return x;
return boss[x]=finds(boss[x]);
}
void Union(ll a,ll b)
{
a=finds(a),b=finds(b);
if(a==b) return;
boss[a]=b;
}
bool yee(ll a,ll b)
{
return dots[a].S>dots[b].S;
}
void dfs(int u,int f)
{
vis[u]=1;
sort(ALL(hav[u]),yee);
ll L=hav[u].size(),U=0,D=hav[u].size()-1,F,S,X=0,Y=0;
ll ucat=-1,dcat=-1;
map<ll,pll> mp;
ans=(ans+(L-1)*L*(L+1)/6)%MOD;
for(ll i:hav[u])
{
Y=(Y+X)%MOD;
for(ll j:G[i])
if(boss[j]!=f)
{
if(!vis[boss[j]]) dfs(boss[j],u);
F=val[j].F,S=val[j].S;
ans=(ans+S*U+((U+1)*(U+2)/2-1)*F+F+S+S*D+((D+1)*(D+2)/2-1)*F)%MOD;
val[i]=(MP(val[i].F+F,(val[i].S+F+S)%MOD));
auto p=mp.find(boss[j]);
if(p!=mp.end())
{
p->S.S+=p->S.F;
X-=p->S.F,Y=(Y-p->S.S+MOD)%MOD;
ans=(ans+F*Y+X*S+X*F)%MOD;
X+=p->S.F,Y=(Y+p->S.S)%MOD;
p->S.F+=F,p->S.S=(p->S.S+F+S)%MOD;
}
else
{
ans=(ans+F*Y+X*S+X*F)%MOD;
mp[boss[j]]=MP(F,F+S);
}
X+=F,Y=(Y+F+S)%MOD;
}
else
{
if(!~ucat) ucat=i;
dcat=i;
}
++U,--D;
}
X=0,Y=0;
for(ll i:hav[u])
if(i!=ucat)
Y=(Y+X)%MOD,X+=val[i].F,Y=(Y+val[i].S)%MOD;
else
{
Y=(Y+X)%MOD,val[i].F+=X,val[i].S=(val[i].S+Y)%MOD;
break;
}
reverse(ALL(hav[u]));
X=0,Y=0;
for(ll i:hav[u])
if(i!=dcat)
Y=(Y+X)%MOD,X+=val[i].F,Y=(Y+val[i].S)%MOD;
else
{
Y=(Y+X)%MOD,val[i].F+=X,val[i].S=(val[i].S+Y)%MOD;
break;
}
}
int DistanceSum(int N, int *X, int *Y)
{
for(int i=0;i<N;++i)
dots[i]=MP(Y[i],X[i]),boss[i]=i,mpp[(ll)Y[i]<<31|X[i]]=i;
for(int i=0;i<N;++i)
for(int j=2;j<4;++j)
{
auto p=mpp.find(((ll)(Y[i]+dx[j])<<31)|(X[i]+dy[j]));
if(p!=mpp.end()) Union(i,p->S);
}
for(int i=0;i<N;++i)
hav[finds(i)].pb(i),val[i]=MP(1,0);
for(int i=0;i<N;++i)
for(int j=0;j<2;++j)
{
auto p=mpp.find(((ll)(Y[i]+dx[j])<<31)|(X[i]+dy[j]));
if(p!=mpp.end()) G[i].pb(p->S);
}
dfs(finds(0),-1);
return ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
5120 KB |
Output is correct |
2 |
Correct |
6 ms |
4992 KB |
Output is correct |
3 |
Correct |
6 ms |
5120 KB |
Output is correct |
4 |
Correct |
6 ms |
4992 KB |
Output is correct |
5 |
Correct |
17 ms |
5120 KB |
Output is correct |
6 |
Correct |
6 ms |
4992 KB |
Output is correct |
7 |
Correct |
6 ms |
5120 KB |
Output is correct |
8 |
Correct |
5 ms |
5120 KB |
Output is correct |
9 |
Correct |
6 ms |
5028 KB |
Output is correct |
10 |
Correct |
6 ms |
5120 KB |
Output is correct |
11 |
Correct |
5 ms |
5120 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
5216 KB |
Output is correct |
2 |
Correct |
6 ms |
5248 KB |
Output is correct |
3 |
Correct |
6 ms |
5248 KB |
Output is correct |
4 |
Correct |
6 ms |
5248 KB |
Output is correct |
5 |
Correct |
7 ms |
5376 KB |
Output is correct |
6 |
Correct |
8 ms |
5376 KB |
Output is correct |
7 |
Correct |
8 ms |
5376 KB |
Output is correct |
8 |
Correct |
7 ms |
5376 KB |
Output is correct |
9 |
Correct |
8 ms |
5376 KB |
Output is correct |
10 |
Correct |
7 ms |
5376 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
24 ms |
7928 KB |
Output is correct |
2 |
Correct |
24 ms |
7820 KB |
Output is correct |
3 |
Correct |
73 ms |
12096 KB |
Output is correct |
4 |
Correct |
58 ms |
11816 KB |
Output is correct |
5 |
Correct |
159 ms |
19028 KB |
Output is correct |
6 |
Correct |
144 ms |
18824 KB |
Output is correct |
7 |
Correct |
160 ms |
19284 KB |
Output is correct |
8 |
Correct |
167 ms |
18952 KB |
Output is correct |
9 |
Correct |
163 ms |
18568 KB |
Output is correct |
10 |
Correct |
140 ms |
18316 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
28 ms |
7940 KB |
Output is correct |
2 |
Correct |
26 ms |
8312 KB |
Output is correct |
3 |
Correct |
64 ms |
12328 KB |
Output is correct |
4 |
Correct |
65 ms |
12584 KB |
Output is correct |
5 |
Correct |
154 ms |
19336 KB |
Output is correct |
6 |
Correct |
174 ms |
19776 KB |
Output is correct |
7 |
Correct |
154 ms |
19660 KB |
Output is correct |
8 |
Correct |
156 ms |
19592 KB |
Output is correct |
9 |
Correct |
159 ms |
19080 KB |
Output is correct |
10 |
Correct |
149 ms |
19080 KB |
Output is correct |