QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#89435#5415. RopewayIanD#WA 2ms3428kbC++143.2kb2023-03-20 05:01:192023-03-20 05:01:22

Judging History

你现在查看的是最新测评结果

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2023-03-20 05:01:22]
  • 评测
  • 测评结果:WA
  • 用时:2ms
  • 内存:3428kb
  • [2023-03-20 05:01:19]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

#define rep(i, a, b) for(int i = a; i < (b); ++i)
#define all(x) begin(x), end(x)
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;

#define pb push_back
#define mp make_pair

#define MAXN 500'010


template<class T>
struct RMQ {
	vector<vector<T>> jmp;
	RMQ(const vector<T>& V) : jmp(1, V) {
		for (int pw = 1, k = 1; pw * 2 <= sz(V); pw *= 2, ++k) {
			jmp.emplace_back(sz(V) - pw * 2 + 1);
			rep(j,0,sz(jmp[k]))
				jmp[k][j] = min(jmp[k - 1][j], jmp[k - 1][j + pw]);
		}
	}
	T query(int a, int b) {
		assert(a < b); // or return inf if a == b
		int dep = 31 - __builtin_clz(b - a);
		return min(jmp[dep][a], jmp[dep][b - (1 << dep)]);
	}
};

int n, k;
ll a[MAXN];
string s;
ll dp[MAXN];
vector<ll> dp2;

int main() {
	ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int t;
    cin >> t;
    while (t--) {
    
    cin >> n >> k;
    dp2.resize(n);
    rep(i, 0, n) cin >> a[i];
    cin >> s;

    priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> q;
    q.push(mp(0, -1));
    rep(i, 0, n) {
        while (q.top().second < i-k) q.pop();
        dp[i] = q.top().first + a[i];
        if (s[i] == '1') {
            while (!q.empty()) q.pop();
        }
        q.push(mp(dp[i], i));
    }
    while (!q.empty()) q.pop();
    q.push(mp(0, n));
    for (int i = n-1; i >= 0; i--) {
        while (q.top().second > i+k) q.pop();
        dp2[i] = q.top().first + a[i];
        if (s[i] == '1') {
            while (!q.empty()) q.pop();
        }
        q.push(mp(dp2[i], i));
    }

    RMQ<ll> rt(dp2);

    int numq;
    cin >> numq;
    //cout << numq << endl;
    while (numq--) {
        int ind;
        ll newval;
        cin >> ind >> newval;
        ind--;

        ll includecost = newval;
        {
        ll leftcost = 0;
        if (ind >= k) leftcost = dp[ind-1];
        for (int i = ind-1; i >= max(0, ind-k); i--) {
            leftcost = min(leftcost, dp[i]);
            if (s[i] == '1') break;
        }

        ll rightcost = 0;
        if (ind < n-k) rightcost = dp2[ind+1];
        for (int i = ind+1; i <= min(n-1, ind+k); i++) {
            rightcost = min(rightcost, dp2[i]);
            if (s[i] == '1') break;
        }

        includecost += rightcost + leftcost;
        }

        ll excludecost = -1;
        {
        int nextrights = n;
        for (int i = ind+1; i <= min(n-1, ind+k); i++) {
            if (s[i] == '1') {
                nextrights = i;
                break;
            }
        }

        for (int i = ind-1; i >= max(-1, ind-k+1); i--) {
            ll med = 0;
            if (i != -1) med = dp[i];
            if (i < n-k || nextrights < n) med += rt.query(ind+1, min(nextrights+1, i+k+1)); // include something to the right
            if (excludecost == -1 || med < excludecost) excludecost = med;
        }
        }
        if (s[ind] == '1' || excludecost == -1) excludecost = includecost;
        cout << min(excludecost, includecost) << '\n';
        //cout << excludecost << ' ' << includecost << endl;

        
    }


    }
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 3428kb

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: 2ms
memory: 3392kb

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:

1984533677
1423635430
2593321629
2364998688
1541797430
2223491101
2285479513
1765265336
2494894393
2240907690
2577284958
2494894393
2766700195
2511807344
2494894393
1239870630
2669077215
2682112324
27557517
470735446
211705888
564509290
390951726
470735446
314524771
15
19
19
19
20
18
54
15465560
154...

result:

wrong answer 1st numbers differ - expected: '2472886431', found: '1984533677'