CSES - Datatähti 2025 alku - Results
Submission details
Task:Robotti
Sender:OnniHak
Submission time:2024-10-31 00:11:37 +0200
Language:C++ (C++17)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED30
#2ACCEPTED70
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2details
#2ACCEPTED0.00 s1, 2details
#3ACCEPTED0.00 s1, 2details
#4ACCEPTED0.00 s1, 2details
#5ACCEPTED0.00 s1, 2details
#6ACCEPTED0.00 s1, 2details
#7ACCEPTED0.01 s1, 2details
#8ACCEPTED0.00 s1, 2details
#9ACCEPTED0.00 s1, 2details
#10ACCEPTED0.00 s1, 2details
#11ACCEPTED0.00 s1, 2details
#12ACCEPTED0.00 s2details
#13ACCEPTED0.00 s2details
#14ACCEPTED0.01 s2details
#15ACCEPTED0.01 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.01 s2details
#18ACCEPTED0.01 s2details
#19ACCEPTED0.03 s2details
#20ACCEPTED0.03 s2details
#21ACCEPTED0.01 s2details
#22ACCEPTED0.01 s2details
#23ACCEPTED0.04 s2details
#24ACCEPTED0.04 s2details

Compiler report

input/code.cpp: In function 'std::vector<int> getStepsCoins(std::string&)':
input/code.cpp:88:48: warning: variable 'coinIdx' set but not used [-Wunused-but-set-variable]
   88 |     int steps=0, gatheredCoins=0, robotIdxIdx, coinIdx, coinIdxsSize;
      |                                                ^~~~~~~

Code

#include <iostream>
#include <string>
#include <vector>


int getDistance(int idx1, int idx2);

std::vector<int> getStepsClosestCoinIdx(
    const std::vector<int>& coinIdxs, 
    std::vector<std::vector<int>>& coinIdxP,
    int& robotIdxIdx);

std::vector<int> getStepsCoins(std::string& roomSeries);


int main() {
    int roomAmount; // perhaps the most useless var of this century
    std::string roomSeries;

    
    std::cin >> roomAmount;
    std:: cin >> roomSeries;


    std::vector<int> stepsCoins = getStepsCoins(roomSeries);

    std::cout << stepsCoins.at(0) << " " << stepsCoins.at(1);

    return 0;
}



int getDistance(int idx1, int idx2) 
{
    return abs(idx2 - idx1);
}


// most intuitive c++ function here...
std::vector<int> getStepsClosestCoinIdx(
    const std::vector<int>& coinIdxs, 
    std::vector<std::vector<int>>& coinIdxP,
    int& robotIdxIdx
    ) 
{
    int dist1, dist2, coinIdx1, coinIdx2;

    
    if (coinIdxP.at(robotIdxIdx).at(0) == -1) {
        coinIdx2 = coinIdxs.at(coinIdxP.at(robotIdxIdx).at(1));
        dist2 = getDistance(coinIdx2, coinIdxs.at(robotIdxIdx));
        coinIdxP.at(coinIdxP.at(robotIdxIdx).at(1)).at(0) = coinIdxP.at(robotIdxIdx).at(0);
        robotIdxIdx = coinIdxP.at(robotIdxIdx).at(1);
        return {dist2, coinIdx2};
    }

    if (coinIdxP.at(robotIdxIdx).at(1) ==  static_cast<int>(coinIdxs.size())) {
        coinIdx1 = coinIdxs.at(coinIdxP.at(robotIdxIdx).at(0));
        dist1 = getDistance(coinIdx1, coinIdxs.at(robotIdxIdx));
        coinIdxP.at(coinIdxP.at(robotIdxIdx).at(0)).at(1) = coinIdxP.at(robotIdxIdx).at(1);
        robotIdxIdx = coinIdxP.at(robotIdxIdx).at(0);
        return {dist1, coinIdx1};
    }
    
    coinIdx1 = coinIdxs.at(coinIdxP.at(robotIdxIdx).at(0));
    coinIdx2 = coinIdxs.at(coinIdxP.at(robotIdxIdx).at(1));
    dist1 = getDistance(coinIdx1, coinIdxs.at(robotIdxIdx));
    dist2 = getDistance(coinIdx2, coinIdxs.at(robotIdxIdx));
    coinIdxP.at(coinIdxP.at(robotIdxIdx).at(1)).at(0) = coinIdxP.at(robotIdxIdx).at(0);
    coinIdxP.at(coinIdxP.at(robotIdxIdx).at(0)).at(1) = coinIdxP.at(robotIdxIdx).at(1);

    if (dist1 == dist2) return {dist1, coinIdx1, coinIdx2};

    if (dist1 < dist2) {
        robotIdxIdx = coinIdxP.at(robotIdxIdx).at(0);
        return {dist1, coinIdx1};
    }

    robotIdxIdx = coinIdxP.at(robotIdxIdx).at(1);
    return {dist2, coinIdx2};
}

std::vector<int> getStepsCoins(std::string& roomSeries) {
    int roomAmount = static_cast<int>(roomSeries.size());
    if (roomAmount == 1) return {0, 0};

    int steps=0, gatheredCoins=0, robotIdxIdx, coinIdx, coinIdxsSize;

    std::vector<int> stepsCoinIdx, coinIdxs;
    std::vector<std::vector<int>> coinIdxP;

    for (int i=0; i<roomAmount; i++) {
        if (roomSeries[i] == '.') continue;

        coinIdxsSize = static_cast<int>(coinIdxs.size());

        if (roomSeries[i] == '*') {
            coinIdxs.push_back(i);
            coinIdxP.push_back({coinIdxsSize-1, coinIdxsSize+1});
        } else if (roomSeries[i] == 'R') {
            robotIdxIdx = coinIdxs.size(); // robotti sijaitsee täällä på coinIdxs ja coinIdxP listoissa :)
            coinIdxs.push_back(i);
            coinIdxP.push_back({coinIdxsSize-1, coinIdxsSize+1});
        }
    }

    int coinsTotalAmount =  static_cast<int>(coinIdxs.size())-1;

    for (int i=0; i<coinsTotalAmount; i++) {
        stepsCoinIdx = getStepsClosestCoinIdx(coinIdxs, coinIdxP, robotIdxIdx);
        
        coinIdx = stepsCoinIdx.at(1);
        

        if (stepsCoinIdx.size() > 2) break;

        steps += stepsCoinIdx.at(0);
        gatheredCoins++;

        if (gatheredCoins >= coinsTotalAmount) break;
    }

    return {steps, gatheredCoins};
}


Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
1
R

correct output
0 0

user output
0 0

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
10
...R......

correct output
0 0

user output
0 0

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
10
**.R...***

correct output
12 5

user output
12 5

Test 4

Group: 1, 2

Verdict: ACCEPTED

input
10
***R******

correct output
0 0

user output
0 0

Test 5

Group: 1, 2

Verdict: ACCEPTED

input
1000
R................................

correct output
947 9

user output
947 9

Test 6

Group: 1, 2

Verdict: ACCEPTED

input
1000
.................................

correct output
886 9

user output
886 9

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
1000
.....*..*....**..**..*......*....

correct output
1287 400

user output
1287 400

Test 8

Group: 1, 2

Verdict: ACCEPTED

input
1000
************.*****************...

correct output
0 0

user output
0 0

Test 9

Group: 1, 2

Verdict: ACCEPTED

input
1000
******************************...

correct output
0 0

user output
0 0

Test 10

Group: 1, 2

Verdict: ACCEPTED

input
1000
R*****************************...

correct output
999 999

user output
999 999

Test 11

Group: 1, 2

Verdict: ACCEPTED

input
1000
******************************...

correct output
999 999

user output
999 999

Test 12

Group: 2

Verdict: ACCEPTED

input
10000
.......**........*...........*...

correct output
10971 999

user output
10971 999

Test 13

Group: 2

Verdict: ACCEPTED

input
10000
*..*....*......*.....*..*........

correct output
9999 999

user output
9999 999

Test 14

Group: 2

Verdict: ACCEPTED

input
10000
*.*.*...**.*...*....**.**.**.....

correct output
18766 5000

user output
18766 5000

Test 15

Group: 2

Verdict: ACCEPTED

input
10000
R*****************************...

correct output
9999 9999

user output
9999 9999

Test 16

Group: 2

Verdict: ACCEPTED

input
10000
******************************...

correct output
9999 9999

user output
9999 9999

Test 17

Group: 2

Verdict: ACCEPTED

input
200000
.................................

correct output
0 0

user output
0 0

Test 18

Group: 2

Verdict: ACCEPTED

input
200000
.................................

correct output
299934 10000

user output
299934 10000

Test 19

Group: 2

Verdict: ACCEPTED

input
200000
**.***....**..**.....***.*..*....

correct output
299998 100000

user output
299998 100000

Test 20

Group: 2

Verdict: ACCEPTED

input
200000
******************************...

correct output
0 0

user output
0 0

Test 21

Group: 2

Verdict: ACCEPTED

input
200000
R................................

correct output
133765 3

user output
133765 3

Test 22

Group: 2

Verdict: ACCEPTED

input
200000
R................................

correct output
199982 5000

user output
199982 5000

Test 23

Group: 2

Verdict: ACCEPTED

input
200000
R*****************************...

correct output
199999 199999

user output
199999 199999

Test 24

Group: 2

Verdict: ACCEPTED

input
200000
******************************...

correct output
199999 199999

user output
199999 199999