Submission #912879

#	Time	Username	Problem	Language	Result	Execution time	Memory
912879		Muhammad_Aneeq	Ideal city (IOI12_city)	C++17	44 / 100	1028 ms	8272 KiB

city

#include <cmath>
#include <vector>
#include <map>
#include <algorithm>
#include <queue>
#include <iostream>
using namespace std;
long long const MAXN=1e5+10;
vector<long long>nei[MAXN]={};
long long mod=1e9;
long long ans=0;
vector<long long>dx={0,0,-1,1},dy={-1,1,0,0};
long long dis[MAXN]={};
bool vis[MAXN]={};
void bfs(long long x)
{
	priority_queue<pair<long long,long long>,vector<pair<long long,long long>>,greater<pair<long long,long long>>>pq;
	pq.push({0,x});
	dis[x]=0;
	while (pq.size())
	{
		auto y=pq.top();
		pq.pop();
		if (vis[y.second])
			continue;
		vis[y.second]=1;
		for (auto j:nei[y.second])
		{
			if (!vis[j])
			{
				dis[j]=y.first+1;
				pq.push({dis[j],j});
			}
		}
	}
}
int DistanceSum(int N, int X[], int Y[])
{
	map<long long,vector<long long>>d;
	map<pair<long long,long long>,long long>ex,cnt;
	for (long long i=0;i<N;i++)
		d[X[i]].push_back(Y[i]);
	bool subtask_3=1;
	for (auto& i:d)
	{
		sort(begin(i.second),end(i.second));
		if (i.second.size()!=i.second.back()-i.second[0]+1)
		{
			subtask_3=0;
			break;
		}
	}
	if (subtask_3)
	{
		sort(Y,Y+N);
		sort(X,X+N);
		long long ans=0;
		long long szx=0,sux=0,szy=0,suy=0;
		for (long long i=0;i<N;i++)
		{
			long long g=1ll*Y[i]*szy-suy;
			g%=mod;
			ans=(ans+g)%mod;
			g=1ll*X[i]*szx-sux;
			g%=mod;
			ans=(ans+g)%mod;
			szy++;
			suy+=Y[i];
			szx++;
			sux+=X[i];
		}
		return ans;
	}
	for (int i=0;i<N;i++)
	{
		for (int j=0;j<N;j++)
		{
			if (abs(X[i]-X[j])+abs(Y[i]-Y[j])==1)
				nei[i].push_back(j);
		}
	}
	for (long long i=0;i<N;i++)
		dis[i]=0;
	for (long long i=0;i<N;i++)
	{
		bfs(i);
		for (long long j=i+1;j<N;j++)
		{
			dis[j]%=mod;
			ans=(ans+dis[j])%mod;
		}
		for (long long j=0;j<N;j++)
			dis[j]=vis[j]=0;
	}
	return ans;
}

Compilation message (stderr)

city.cpp: In function 'int DistanceSum(int, int*, int*)':
city.cpp:48:22: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} [-Wsign-compare]
   48 |   if (i.second.size()!=i.second.back()-i.second[0]+1)
      |       ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 21 / 21
• Subtask 3: 23 / 23
• Subtask 4: 0 / 45