oj.uz

Submission #334283

# Submission time Handle Problem Language Result Execution time Memory
334283 2020-12-08T23:31:58 Z HoneyBadger Growing Vegetable is Fun 3 (JOI19_ho_t3) C++17
15 / 100
 149 ms 202752 KB 
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <vector>
#include <bitset>
#include <string>
#include <cstring>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <deque>
#include <utility>
#include <algorithm>
#include <random>
#include <cmath>
#include <cassert>
#include <climits>
#include <ctime>
#include <chrono>



#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,sse4.1,sse4.2,popcnt,abm,mmx,avx,avx2,tune=native")



#ifdef LOCAL
    #define dbg(x) cout << #x << " : " << x << endl;
#else
    #define dbg(x)
#endif


#define pb push_back
#define ppb pop_back()
#define mp make_pair
#define fi(a, b) for (int i = a; i < b; i++)
#define fj(a, b) for (int j = a; j < b; j++)
#define fk(a, b) for (int k = a; k < b; k++)
#define fi1(a, b) for (int i = a - 1; i >= b; i--)
#define fj1(a, b) for (int j = a - 1; j >= b; j--)
#define fk1(a, b) for (int k = a - 1; k >= b; k--)
#define fx(x, a) for (auto& x : a)
#define rep(i, a, b) for (int i = a; i < b; ++i)
#define rep1(i, a, b) for (int i = a - 1; i >= b; --i)
#define siz(x) (int)x.size()
#define lb lower_bound
#define ub upper_bound
#define all(x) x.begin(), x.end()

using namespace std;

template<typename T1, typename T2>inline void mine(T1 &x, const T2 &y) { if (y < x) x = y; }
template<typename T1, typename T2>inline void maxe(T1 &x, const T2 &y) { if (x < y) x = y; }

ostream& operator << (ostream &out, const vector<int> &b) {
    for (auto k : b) out << k << ' ';
    return out;
}

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef char ch;
typedef string str;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;
typedef vector<ch> vch;
typedef vector<vch> vvch;
typedef vector<str> vs;



const int MOD = 1000000007;
const int INF = 1000000050;
const long long BIG = (long long)2e18 + 50;
const int MX = 410;
const double EPS = 1e-9;


mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int n;
int cnt[MX][3];

int dp[MX][MX / 2][MX / 2][3];  // pref cur0 cur1 last
int p[3];
int np[3];

void no() {
	cout << "-1\n";
	exit(0);
}
int32_t main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    cin >> n;
    vi a(n);
    fx(x, a) {
    	char c;
    	cin >> c;
    	if (c == 'R')
    		x = 0;
    	else if (c == 'G')
    		x = 1;
    	else
    		x = 2;
    }
    fi(1, n + 1) {
    	fj(0, 3) {
    		cnt[i][j] = cnt[i - 1][j]; 
    	}
    	cnt[i][a[i - 1]]++;
    }
    fj(0, 3) {
    	if (cnt[n][j] > (n + 1) / 2) {
    		no();
    	}
    }
 	int c0 = cnt[n][0];
 	int c1 = cnt[n][1];
 	int c2 = cnt[n][2];

    fi(0, MX) fj(0, MX / 2) fk(0, MX / 2) for (int l = 0; l < 3; ++l) dp[i][j][k][l] = INF;
   	for (int l = 0; l < 3; ++l) dp[0][0][0][l] = 0;
    fi(0, n) {
    	int b0 = min(c0, i);
    	int b1 = min(c1, i);
    	for (p[0] = 0; p[0] <= b0; ++p[0]) {
    		for (p[1] = 0; p[1] <= b1; ++p[1]) {
    			p[2] = i - p[0] - p[1];
    			if (p[2] < 0) break;
    			if (p[2] > c2) continue;
    			memcpy(np, p, sizeof(p));
    			for (int l = 0; l < 3; ++l) {
    				if (i && !p[l]) continue;
    				for (int l2 = 0; l2 < 3; ++l2) {
    					if (l2 == l) continue;
    					np[l2]++;
    					int delta = 0;
    					for (int col = 0; col < 3; ++col)
    						delta += abs(np[col] - cnt[i + 1][col]);
    					mine(dp[i + 1][np[0]][np[1]][l2], dp[i][p[0]][p[1]][l] + delta);
    					np[l2]--;
    				}
    			} 
    		}
    	}
    }
    int ans = INF;
    for (int l = 0; l < 3; ++l) {
    	mine(ans, dp[n][c0][c1][l] / 2);
    }
    cout << ans << '\n';
}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 112 ms 202600 KB Output is correct
2 Correct 112 ms 202708 KB Output is correct
3 Correct 108 ms 202732 KB Output is correct
4 Correct 108 ms 202604 KB Output is correct
5 Correct 108 ms 202752 KB Output is correct
6 Correct 109 ms 202604 KB Output is correct
7 Correct 110 ms 202732 KB Output is correct
8 Correct 108 ms 202652 KB Output is correct
9 Correct 106 ms 202624 KB Output is correct
10 Correct 1 ms 364 KB Output is correct
11 Incorrect 109 ms 202604 KB Output isn't correct
12 Halted 0 ms 0 KB -

Subtask #2
0 / 55.0

# Verdict Execution time Memory Grader output
1 Correct 112 ms 202600 KB Output is correct
2 Correct 112 ms 202708 KB Output is correct
3 Correct 108 ms 202732 KB Output is correct
4 Correct 108 ms 202604 KB Output is correct
5 Correct 108 ms 202752 KB Output is correct
6 Correct 109 ms 202604 KB Output is correct
7 Correct 110 ms 202732 KB Output is correct
8 Correct 108 ms 202652 KB Output is correct
9 Correct 106 ms 202624 KB Output is correct
10 Correct 1 ms 364 KB Output is correct
11 Incorrect 109 ms 202604 KB Output isn't correct
12 Halted 0 ms 0 KB -

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 122 ms 202680 KB Output is correct
2 Correct 137 ms 202604 KB Output is correct
3 Correct 138 ms 202688 KB Output is correct
4 Correct 138 ms 202604 KB Output is correct
5 Correct 149 ms 202604 KB Output is correct
6 Correct 136 ms 202604 KB Output is correct
7 Correct 148 ms 202604 KB Output is correct
8 Correct 137 ms 202604 KB Output is correct
9 Correct 136 ms 202604 KB Output is correct
10 Correct 135 ms 202604 KB Output is correct
11 Correct 135 ms 202604 KB Output is correct
12 Correct 113 ms 202604 KB Output is correct
13 Correct 115 ms 202676 KB Output is correct
14 Correct 123 ms 202604 KB Output is correct
15 Correct 1 ms 364 KB Output is correct
16 Correct 1 ms 364 KB Output is correct

Subtask #4
0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 112 ms 202600 KB Output is correct
2 Correct 112 ms 202708 KB Output is correct
3 Correct 108 ms 202732 KB Output is correct
4 Correct 108 ms 202604 KB Output is correct
5 Correct 108 ms 202752 KB Output is correct
6 Correct 109 ms 202604 KB Output is correct
7 Correct 110 ms 202732 KB Output is correct
8 Correct 108 ms 202652 KB Output is correct
9 Correct 106 ms 202624 KB Output is correct
10 Correct 1 ms 364 KB Output is correct
11 Incorrect 109 ms 202604 KB Output isn't correct
12 Halted 0 ms 0 KB -