QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#493507#5415. Ropeway333zhanWA 3ms3644kbC++202.9kb2024-07-27 09:59:382024-07-27 09:59:39

Judging History

This is the latest submission verdict.

  • [2024-07-27 09:59:39]
  • Judged
  • Verdict: WA
  • Time: 3ms
  • Memory: 3644kb
  • [2024-07-27 09:59:38]
  • Submitted

answer

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int inf = 1E18;

inline int read () {
    int w = 1, s = 0; char ch = getchar ();
    for (; ! isdigit (ch); ch = getchar ()) if (ch == '-') w = -1;
    for (; isdigit (ch); ch = getchar ()) s = (s << 1) + (s << 3) + (ch ^ 48);
    return s * w;
}

void solve () {
	int n, k;
    cin >> n >> k;
    
    vector <int> a (n + 2);
    for (int i = 1; i <= n; i ++) {
        cin >> a[i];
    }

    string s;
    cin >> s;

    s = " " + s + "0";

    //f从前往后,g从后往前
    vector <int> f (n + 2), g (n + 2);
    
    auto work1 = [&] (vector <int> &dp) {
        deque <int> q;
        q.push_back (0);
        for (int i = 1; i <= n + 1; i ++) {
            while (q.size () && q.front () < i - k) {
                q.pop_front ();
            }
            dp[i] = dp[q.front ()] + a[i];
            if (s[i] == '1') {
                q.clear ();
            }
            while (q.size () && dp[q.back ()] >= dp[i]) {
                q.pop_back ();
            }
            q.push_back (i);
        }
    };

    work1 (f);
    reverse (s.begin () + 1, s.end ());
    reverse (a.begin () + 1, a.end ());
    work1 (g);
    reverse (s.begin () + 1, s.end ());
    reverse (a.begin () + 1, a.end ());
    reverse (g.begin () + 1, g.end ());
    
    for (int i = 1; i <= n; i ++) {
        g[i] -= a[i];
    }

    vector <int> h (n + 2);
    auto work2 = [&] (int x, int y) -> int {
        int ans = inf;
        int old = a[x];
        a[x] = y;

        deque <int> q;
        for (int i = k; i > 0; i --) {
            if (x - i >= 0) {
                if (s[x - i] == '1') {
                    q.clear ();
                }
                while (q.size () && f[q.back ()] >= f[x - i]) {
                    q.pop_back ();
                }
                q.push_back (x - i);
            }
        }
        for (int i = x; i < x + k && i <= n + 1; i ++) {
            while (q.size () && q.front () < i - k) {
                q.pop_front ();
            }
            if (q.front () < x) {
                h[i] = f[q.front ()] + a[i];
            } else {
                h[i] = h[q.front ()] + a[i];
            }
            ans = min (ans, h[i] + g[i]);
            if (s[i] == '1') {
                q.clear ();
            }
            while (q.size () && (q.back () < x ? f[q.back ()] : h[q.back ()]) >= h[i]) {
                q.pop_back ();
            }
            q.push_back (i);
        }
        return ans;
    };

    int q;
    cin >> q;

    while (q --) {
        int p, v;
        cin >> p >> v;

        cout << work2 (p, v) << '\n';
    }
} 

signed main () {
	ios::sync_with_stdio (false);
    cin.tie (nullptr);
    
	int T = 1; 
	cin >> T;
	// T = read ();

	while (T --) {
		solve ();
	}
	
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3600kb

input:

3
10 3
5 10 7 100 4 3 12 5 100 1
0001000010
2
2 3
6 15
5 6
1 1 1 1 1
00000
1
3 100
5 6
1 1 1 1 1
00100
1
3 100

output:

206
214
0
100

result:

ok 4 number(s): "206 214 0 100"

Test #2:

score: -100
Wrong Answer
time: 3ms
memory: 3644kb

input:

500
19 6
285203460 203149294 175739375 211384407 323087820 336418462 114884618 61054702 243946442 19599175 51974474 285317523 222489944 26675167 300331960 1412281 324105264 33722550 169011266
1111111110110100011
18
3 127056246
5 100630226
14 301161052
2 331781882
5 218792226
2 190274295
12 49227476
...

output:

2472886431
2299111966
2796055445
2379061425
2417273966
2355715838
2285479513
2285479513
2521569560
2075053968
2341194911
2521569560
2530610148
2345953622
2521569560
2438434986
2669077215
2682112324
470735446
470735446
211705888
564509290
715543137
470735446
564509290
18
18
18
18
20
19
54
849950346
8...

result:

wrong answer 4th numbers differ - expected: '2650202148', found: '2379061425'