Submission #370093

#	Time	Username	Problem	Language	Result	Execution time	Memory
370093		AriaH	Raspad (COI17_raspad)	C++11	0 / 100	1038 ms	82284 KiB

raspad

/** Im the best because i work as hard as i possibly can **/

#pragma GCC optimize("O2")
#include <bits/stdc++.h>
using namespace std;

typedef long long                   ll;
typedef long double                 ld;
typedef pair<int,int>               pii;
typedef pair<ll,ll>                 pll;
#define all(x)                      (x).begin(),(x).end()
#define F                           first
#define S                           second
#define Mp                          make_pair
#define fast_io                     ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file_io                     freopen("in.txt" , "r+" , stdin) ; freopen("out.txt" , "w+" , stdout);
#define endl                        "\n"
#define SZ(x)						(int)x.size()

const int N = 1e5 + 10;
const ll mod = 1e9 + 7;
const ll mod2 = 998244353;
const ll inf = 8e18;
const int LOG = 22;
const int M = 52;

char C[N];

int n, m, A[N][M], I[N][M], par[N * M], arr[N][M];

pii RI[N * M];

ll tot, ps[N], cnt[N];

int get(int x) { return ((x == par[x])? x : par[x] = get(par[x])); }

inline int ok(int i)
{
	return A[RI[i].F][RI[i].S];
}

int unify(int v, int u)
{
	if(!ok(v) || !ok(u)) return 0;
	v = get(v), u = get(u);
	if(v == u) return 0;
	if(u < v) swap(u, v);
	par[u] = v;
	return 1;
}

void solve(int l, int r)
{
	if(l == r)
	{
		ll cu = 0;
		for(int j = 1; j <= m; j ++)
		{
			if(A[l][j]) tot ++, cu ++;
			if(A[l][j] && A[l][j + 1]) tot --, cu --;
		}
		///printf("l = %d r = %d cu = %lld\n", l, r, cu);
		return;
	}
	int mid = (l + r) >> 1;
	solve(l, mid); solve(mid + 1, r);
	///printf("l = %d r = %d\n", l, r);
	ps[l - 1] = ps[r + 1] = 0;
	for(int i = l; i <= r; i ++)
	{
		for(int j = 1; j <= m; j ++)
		{
			ps[i] = 0;
			par[I[i][j]] = I[i][j];
			cnt[i] = 0;
		}
	}
	vector < int > merges;
	merges.push_back(mid + 1);
	///printf("show arr : \n");
	for(int i = mid + 1; i <= r; i ++)
	{
		if(i > mid + 1) cnt[i] = cnt[i - 1];
		for(int j = 1; j <= m; j ++)
		{
			cnt[i] += A[i][j];
			cnt[i] -= unify(I[i][j], I[i][j - 1]);
		}
		for(int j = 1; j <= m; j ++) arr[i][j] = get(I[mid + 1][j]);/// printf("%d ", arr[i][j]);
		///printf("\n");
		if(i > mid + 1)
		{
			for(int j = 1; j <= m; j ++)
			{
				cnt[i] -= unify(I[i][j], I[i - 1][j]);
			}
			int flag = 0;
			for(int j = 1; j <= m; j ++)
			{
				if(arr[i][j] != arr[i - 1][j]) flag = 1;
			}
			if(flag) merges.push_back(i);
		}
		ps[i] = ps[i - 1] + cnt[i];
		///printf("phase 1, i = %d cnt = %lld ps = %lld", i, cnt[i], ps[i]);
	}
	///printf("\n");
	merges.push_back(r + 1);
	/*printf("phase 2\n");
	for(auto x : merges)
	{
		printf("%d ", x);
	}
	printf("\n");
	*/
	ll cu = 0;
	for(int i = mid; i >= l; i --)
	{
		for(int j = 1; j <= m; j ++)
		{
			cu += A[i][j];
			cu -= unify(I[i][j], I[i][j - 1]);
		}
		if(i < mid)
		{
			for(int j = 1; j <= m; j ++)
			{
				cu -= unify(I[i][j], I[i + 1][j]);
			}
		}
		///printf("phase 3 i = %d cu = %lld\n", i, cu);
		for(int id = 0; id < SZ(merges) - 1; id ++)
		{
			map < pii , int > mp;
			int yal = 0;
			for(int j = 1; j <= m; j ++)
			{
				if(A[mid][j] + A[mid + 1][j] < 2) continue;
				int v = get(I[mid][j]), u = arr[merges[id]][j];
				///printf("wtf is wrong v = %d u = %d\n", v, u);
				if(mp.find(Mp(v, u)) == mp.end()) yal ++;
				mp[Mp(v, u)] = 1;
				mp[Mp(u, v)] = 1;
			}
			///printf("id = %d yal = %d\n", id, yal);
			tot += (cu - yal) * (merges[id + 1] - merges[id]) + ps[merges[id + 1] - 1] - ps[merges[id] - 1];
		}
	}
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i ++)
	{
		for(int j = 1; j <= m; j ++)
		{
			I[i][j] = (i - 1) * m + j;
			RI[I[i][j]] = Mp(i, j);
		}
	}
	for(int i = 1; i <= n; i ++)
	{
		scanf("%s", C);
		for(int j = 0; j < m; j ++)
		{
			A[i][j + 1] = (C[j] == '1');
		}
	}
	solve(1, n);
	printf("%lld", tot);
    return 0;
}

Compilation message (stderr)

raspad.cpp: In function 'int main()':
raspad.cpp:153:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  153 |  scanf("%d%d", &n, &m);
      |  ~~~~~^~~~~~~~~~~~~~~~
raspad.cpp:164:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  164 |   scanf("%s", C);
      |   ~~~~~^~~~~~~~~

• Subtask 1: 0 / 9
• Subtask 2: 0 / 17
• Subtask 3: 0 / 35
• Subtask 4: 0 / 39