제출 #421203

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
421203		MohamedAhmed04	Rectangles (IOI19_rect)	C++14	10 / 100	285 ms	496132 KiB

rect

#include <bits/stdc++.h>
#include "rect.h"
//#include "grader.cpp"

using namespace std ;

const int MAX = 2500 + 10 ;

struct BIT
{
	vector<int>bit , v ;
	int n ;

	int getidx(int x)
	{
		int idx = upper_bound(v.begin() , v.end() , x) - v.begin() ;
		return idx ;
	}

	void Add(int i , int x)
	{
		for(; i <= n ; i += (i & (-i)))
			bit[i] += x ;
	}

	void Insert(int i)
	{
		Add(getidx(i) , 1) ;
	}

	void Erase(int i)
	{
		Add(getidx(i) , -1) ;
	}

	int get(int i)
	{
		int sum = 0 ;
		for(; i > 0 ; i -= (i & (-i)))
			sum += bit[i] ;
		return sum ;
	}

	int query(int i)
	{
		return get(getidx(i)) ;
	}

	void init(vector<int>&v2)
	{
		n = v2.size() ;
		v = v2 ;
		sort(v.begin() , v.end()) ;
		bit = vector<int>(n+1 , 0) ;
		for(auto &i : v)
			Insert(i) ;
	}
};

BIT tree[MAX][MAX] ;
set<int>candi[MAX] ;
stack<int>st[MAX] ;
vector<int>v2[MAX][MAX] ;

int n , m ;

int calc(int i , int j , int i2 , int last)
{
	int j2 = j-1 ;
	i2++ ;
	if(j2 < 0 || i2 >= i)
		return 0ll ;
	return (tree[i2][j2].query(last+1) - tree[i2][j2].query(j)) ;
}

vector< vector<int> >a ;
vector<int>erased ;

void AddRow(int i)
{
	stack<int>st2 ;
	for(int j = m-1 ; j >= 0 ; --j)
	{
		while(st2.size() && a[i][j] > a[i][st2.top()])
			v2[i][j].push_back(st2.top()) , st2.pop() ;
		if(st2.size())
			v2[i][j].push_back(st2.top()) ;
		while(st2.size() && a[i][st2.top()] == a[i][j])
			st2.pop() ;
		st2.push(j) ;
		sort(v2[i][j].begin() , v2[i][j].end()) ;
		tree[i][j].init(v2[i][j]) ;
	}
	if(!i)
		return ;
	for(int j = 0 ; j < m ; ++j)
	{
		erased.clear() ;
		for(auto &k : v2[i-1][j])
		{
			if(!binary_search(v2[i][j].begin() , v2[i][j].end() , k))
				erased.push_back(k) ;
		}
		for(auto &k : erased)
		{
			for(int i2 = i-1 ; i2 >= 0 ; --i2)
			{
				if(!binary_search(v2[i][j].begin() , v2[i][j].end() , k))
					break ;
				tree[i2][j].Erase(k) ;
			}
		}
	}
}

long long count_rectangles(std::vector<std::vector<int> > v) 
{
	a = v ;
	n = a.size() , m = a[0].size() ;
	long long ans = 0 ;
	for(int i = 0 ; i < n ; ++i)
	{
		for(int j = 0 ; j < m ; ++j)
		{
			while(st[j].size() && a[i][j] > a[st[j].top()][j])
				candi[j].insert(st[j].top()) , st[j].pop() ;
			if(st[j].size())
				candi[j].insert(st[j].top()) ;
			while(st[j].size() && a[st[j].top()][j] == a[i][j])
				st[j].pop() ;
			st[j].push(i) ;
		}
		for(int j = 0 ; j < m ; ++j)
		{
			for(auto &i2 : candi[j])
			{
				int last = j ;
				for(int k = j ; k < m ; ++k)
				{
					if(candi[k].find(i2) == candi[k].end())
						break ;
					last = k ;
				}
				for(int k = j ; k <= last ; ++k)
				{
					ans += calc(i , k , i2 , last) ;
					if(k != j)
						candi[k].erase(i2) ;
				}
			}
			candi[j].clear() ;
		}
		AddRow(i) ;
	}
	return ans ;
}

• Subtask 1: 0 / 8
• Subtask 2: 0 / 7
• Subtask 3: 0 / 12
• Subtask 4: 0 / 22
• Subtask 5: 10 / 10
• Subtask 6: 0 / 13
• Subtask 7: 0 / 28