Submission #145532

# Submission time Handle Problem Language Result Execution time Memory
145532 2019-08-20T11:02:50 Z shashwatchandra Bob (COCI14_bob) C++17
24 / 120
956 ms 22908 KB
/*input
5 3
2 2 2
2 2 1
1 1 1
2 1 2
1 2 1
*/
#include<bits/stdc++.h>
using namespace std;

#define int long long
#define pb push_back
#define REP(i,n)for(int i = 0;i < n;i++)
#define RE(i,n) for(int i = 1;i <= n;i++)
#define f first
#define s second
#define pii pair<int,int> 

const int N = 1001;
int a[N][N];
int eq[N][N];
int seg[4*N];
int lazy[4*N];
bool zero[4*N];
int n,m;

#define child 2*node
#define mid (l+r)/2

inline void pushdown(int node,int l,int r,bool wow = 1){

	seg[node] += (r-l+1)*lazy[node];
	if(l != r){
		lazy[child] += lazy[node];
		lazy[child+1] += lazy[node];
		if(wow){
			pushdown(child,l,mid,0);
			pushdown(child+1,mid+1,r,0);
		}
	}
	lazy[node] = 0;
	if(zero[node]){
		seg[node] = 0;
		if(l != r){
			zero[child] = 1;
			zero[child+1] = 1;
		}
	}
	zero[node] = 0;
}

void assign(int node,int l,int r,int start,int end){
	pushdown(node,l,r);
	if(start > r or l > end)return;
	if(start <= l and r <= end){
		zero[node] = 1;
		pushdown(node,l,r);
		return;
	}
	assign(child,l,mid,start,end);
	assign(child+1,mid+1,r,start,end);
	seg[node] = seg[child]+seg[child+1];
}

void add(int node,int l,int r,int start,int end){
	pushdown(node,l,r);
	if(start > r or l > end)return;
	if(start <= l and r <= end){
		lazy[node]++;
		pushdown(node,l,r);
		return;
	}
	add(child,l,mid,start,end);
	add(child+1,mid+1,r,start,end);
	seg[node] = seg[child]+seg[child+1];
}

void print(int node,int l,int r){
		cout << "FOR : " << node << " " << l << " " << r << endl;
		cout << seg[node] << " " << lazy[node] << " " << zero[node] << endl;
		if(l == r)return;
		print(child,l,mid);
		print(child+1,mid+1,r);
	
}



void solve(){
	cin >> n >> m;
	RE(i,n){
		RE(j,m)cin >> a[i][j];
	}
	RE(i,n){
		int ind = 1;
		RE(j,m){
			if(a[i][j] != a[i][ind]){
				ind = j;
			}
			eq[i][j] = ind;
		}
	}
	int ans = 0;
	RE(col,m){
		RE(row,n){
			if(a[row][col] != a[row-1][col]){
				assign(1,1,m,1,m);
			}
			if(eq[row][col] != 0)assign(1,1,m,1,eq[row][col]-1);
			//for(int i = 1;i < eq[row][col];i++)seg[i] = 0;
			add(1,1,m,eq[row][col],col);
			ans += seg[1];
		}
	}
	cout << ans;
}

signed main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	solve();
}
# Verdict Execution time Memory Grader output
1 Correct 4 ms 760 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 4 ms 760 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 201 ms 8796 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 219 ms 9312 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 204 ms 9436 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 214 ms 9476 KB Output isn't correct
# Verdict Execution time Memory Grader output
1 Incorrect 902 ms 22908 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 930 ms 18348 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 944 ms 18264 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 956 ms 18040 KB Output isn't correct
2 Halted 0 ms 0 KB -