Submission #473053

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
473053	2021-09-14T23:20:53 Z	robell	Bob (COCI14_bob)	C++17	120 / 120	731 ms	29596 KB

#pragma GCC optimize("O2")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
typedef tree<int,null_type,less<int>,rb_tree_tag, tree_order_statistics_node_update> indexed_set;
typedef long long ll;
#define pb push_back
#define eb emplace_back
#define countbits __builtin_popcount
#define beg0 __builtin_clz
#define terminal0 __builtin_ctz
#define mod 1e9+7
void setIO(){
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
}
void setIO(string f){
	freopen((f+".in").c_str(),"r",stdin);
	freopen((f+".out").c_str(),"w",stdout);
	setIO();
}
int g[1000][1000];
pair<ll,ll> dp[1000][1000];
int main(){
	setIO();
	int N, M; cin >> N >> M;
	for (int i=0;i<N;i++){
		for (int j=0;j<M;j++){
			cin >> g[i][j];
		}
	}
	for (int i=0;i<N;i++){
		for (int j=0;j<M;j++){
			dp[i][j].first=1;dp[i][j].second=1;
		}
	}
	for (int i=1;i<N;i++){
		if (g[i-1][0]==g[i][0]){
			dp[i][0].first+=dp[i-1][0].first;
		}
	}
	for (int i=1;i<M;i++){
		if (g[0][i-1]==g[0][i]){
			dp[0][i].second+=dp[0][i-1].second;
		}
	}
	for (int i=1;i<N;i++){
		for (int j=1;j<M;j++){
			if (g[i-1][j]==g[i][j]){
				dp[i][j].first+=dp[i-1][j].first;
			}
			if (g[i][j-1]==g[i][j]){
				dp[i][j].second+=dp[i][j-1].second;
			}
		}
	}
	ll rects = 0;
	for (int i=0;i<N;i++){
		rects+=dp[i][0].first;
	}
	for (int i=0;i<M;i++){
		rects+=dp[0][i].second;
	}
	rects--;
	for (int i=1;i<N;i++){
		for (int j=1;j<M;j++){
			ll length = 1e9;
			for (int z=j;z>j-dp[i][j].second;z--){
				length=min(length,dp[i][z].first);
				rects+=(length);
			}
		}
	}
	cout << rects << "\n";
}

Compilation message

bob.cpp: In function 'void setIO(std::string)':
bob.cpp:19:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   19 |  freopen((f+".in").c_str(),"r",stdin);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
bob.cpp:20:9: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   20 |  freopen((f+".out").c_str(),"w",stdout);
      |  ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	716 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1 ms	716 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	29 ms	8692 KB	Output is correct

Subtask #4
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	51 ms	9072 KB	Output is correct

Subtask #5
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	33 ms	9336 KB	Output is correct

Subtask #6
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	38 ms	9284 KB	Output is correct

Subtask #7
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	138 ms	26436 KB	Output is correct
2	Correct	664 ms	21824 KB	Output is correct
3	Correct	724 ms	21828 KB	Output is correct

Subtask #8
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	160 ms	29596 KB	Output is correct
2	Correct	319 ms	21820 KB	Output is correct
3	Correct	731 ms	21700 KB	Output is correct

Subtask #9
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	227 ms	29348 KB	Output is correct
2	Correct	730 ms	21808 KB	Output is correct
3	Correct	728 ms	21792 KB	Output is correct

Subtask #10
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	185 ms	29508 KB	Output is correct
2	Correct	698 ms	21820 KB	Output is correct
3	Correct	725 ms	21688 KB	Output is correct