Оптимизация решения

@ronny10 · Регистрация: 23.02.2014

Студворк — интернет-сервис помощи студентам

Всем доброго времени суток! Решил задачу:

Имеется ряд из N лампочек, которые пронумерованы от 1 до N. Изначально ни одна из лампочек не горит. Далее происходит K последовательных линейных инверсий этого ряда ламп. Под линейной инверсией понимается инверсия каждой P-й лампочки в ряде. Например, если P=3, то произойдет инверсия 3й, 6й, 9й и т.д. лампочек. Требуется определить: сколько горящих лампочек останется после реализации всех заданных линейных инверсий? (1 ≤ N ≤ 109, 1 ≤ K ≤ 100, 1 ≤ Pi ≤ 50)

ПРОБЛЕМА В СКОРОСТИ РАБОТЫ. Помогите сделать код оптимальнее или решение эффективнее.

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#include "stdafx.h"
#include <vector>
#include <iostream>
#include <map>
#include <iterator>
 
using namespace std;
 
int gcd(int a, int b) {
    int nod = 1L;
    int tmp;
    if (a == 0L)
        return b;
    if (b == 0L)
        return a;
    if (a == b)
        return a;
    if (a == 1L || b == 1L)
        return 1L;
    while (a != 0 && b != 0) {
        if (((a & 1L) | (b & 1L)) == 0L) {
            nod <<= 1L;
            a >>= 1L;
            b >>= 1L;
            continue;
        }
        if (((a & 1L) == 0L) && (b & 1L)) {
            a >>= 1L;
            continue;
        }
        if ((a & 1L) && ((b & 1L) == 0L)) {
            b >>= 1L;
            continue;
        }
        if (a > b) {
            tmp = a;
            a = b;
            b = tmp;
        }
        tmp = a;
        a = (b - a) >> 1L;
        b = tmp;
    }
    if (a == 0)
        return nod * b;
    else
        return nod * a;
}
 
int NOK(int n1, int n2)
{
    return n1 / gcd(n1, n2) * n2;
}
 
int main() {
    int N = 0, K = 0;
    cin >> N;
    cin >> K;
    int* p = new int[K];
    int* pp = new int[K];
    for (int i = 0; i < K; i++)
    {
        cin >> pp[i];
        p[i] = (int)pp[i];
    }
    delete[] pp;
    for (int i = 0; i < K - 1; i++) 
    {
        for (int j = 0; j < K - i - 1; j++) {
            if (p[j] > p[j + 1]) {
                p[j] ^= p[j + 1] ^= p[j] ^= p[j + 1];
            }
        }
    }
    int* p_cleaned = new int[50];
    for (int i = 0; i < 50; i++)
        p_cleaned[i] = 0;
    for (int i = 0; i < K; i++)
        p_cleaned[p[i] - 1]++;
    for (int i = 0; i < 50; i++)
        p_cleaned[i] %= 2;
    vector<int> p_last;
    for (int i = 0; i < 50; i++)
    {
        if (p_cleaned[i] != 0)
            p_last.push_back(i + 1);
    }
    delete[] p;
    delete[] p_cleaned;
    int count = 0;
    map <int , int> arr;
    K = p_last.size();
    arr.insert(pair<int, int>((int)p_last[0], 1));
    for (int i = 1; i < K; i++)
    {
        map<int, int> tmp;
        for (map<int,int>::reverse_iterator riter = arr.rbegin(); riter != arr.rend() ; ++riter)
        {
            int tmp_nok = NOK(riter->first, (int)p_last[i]);
            if (tmp_nok <= N)
            {
                if (tmp.find(tmp_nok) == tmp.end()) {
                    tmp.insert(pair<int, int>(tmp_nok, -2 * riter->second));
                }
                else {
                    tmp.at(tmp_nok) += (-2 * riter->second);
                }
            }   
        }
        if (tmp.find((int)p_last[i]) == tmp.end()) {
            tmp.insert(pair<int, int>((int)p_last[i], 1));
        }
        else {
            tmp.at((int)p_last[i]) += 1;
        }
        for (map<int, int>::iterator iter = tmp.begin(); iter != tmp.end(); iter++)
            if (arr.find(iter->first) == arr.end()) {
                arr.insert(pair<int, int>(iter->first, iter->second));
            }
            else {
                arr.at(iter->first) += iter->second;
            }
//          tmp.erase(tmp.begin(), tmp.end());
        /*for (map<int, int>::iterator iter = arr.begin(); iter != arr.end(); iter++)
            cout << iter->first << " " << iter->second << endl;*/
    }
    for (map<int, int>::iterator iter = arr.begin(); iter != arr.end(); iter++)
        //cout << iter->first << " " << iter->second << endl;
        count += N /  iter->first * iter->second;
 
    cout << count;
    system("pause");
    return 0;
}

@LaHaH · 17.03.2018, 19:30

Если я правильно понял, то вот решение:

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#include <iostream>
#include <vector>
#include <string>
 
void Inverse(std::vector<bool> &vecLightBulbs, int nInverseNumber);
 
int main()
{
    setlocale(LC_ALL, "russian");
 
    std::vector<bool> vecLightBulbs;
 
    std::cout << "Введите количество лампочек:\n";
    int nBulbCount = 0;
    std::cin >> nBulbCount;
 
    for (int i = 0; i < nBulbCount; i++)
        vecLightBulbs.push_back(false);
 
    std::cout << "Введите количество инверсий:\n";
    int nInvCount = 0;
    std::cin >> nInvCount;
 
    std::vector<int> vecInverseParams;
 
    int nInverseNumber = 0;
    for (int i = 0; i < nInvCount; i++)
    {
        std::cout << "Введите инверсию # " + std::to_string(i) << std::endl;
        std::cin >> nInverseNumber;
        vecInverseParams.push_back(nInverseNumber);
 
        Inverse(vecLightBulbs, nInverseNumber);
    }
 
    int nCount = 0;
    for (auto item : vecLightBulbs)
    {
        if (item == true)
            nCount++;
    }
 
    std::cout << "Количество включенных лампочек: " + std::to_string(nCount) << std::endl;
 
    system("pause");
    return 0;
}
 
void Inverse(std::vector<bool> &vecLightBulbs, int nInverseNumber)
{
    for (size_t i = nInverseNumber; i < vecLightBulbs.size(); i += nInverseNumber)
        vecLightBulbs[i] = !vecLightBulbs[i];
}

Новые блоги и статьи Все статьи Все блоги /
Новый ноутбук volvo 07.12.2025 Всем привет. По скидке в "черную пятницу" взял себе новый ноутбук Lenovo ThinkBook 16 G7 на Амазоне: Ryzen 5 7533HS 64 Gb DDR5 1Tb NVMe 16" Full HD Display Win11 Pro	Музыка, написанная Искусственным Интеллектом volvo 04.12.2025 Всем привет. Некоторое время назад меня заинтересовало, что уже умеет ИИ в плане написания музыки для песен, и, собственно, исполнения этих самых песен. Стихов у нас много, уже вышли 4 книги, еще 3. . .	От async/await к виртуальным потокам в Python IndentationError 23.11.2025 Армин Ронахер поставил под сомнение async/ await. Создатель Flask заявляет: цветные функции - провал, виртуальные потоки - решение. Не threading-динозавры, а новое поколение лёгких потоков. Откат?. . .	Поиск "дружественных имён" СОМ портов Argus19 22.11.2025 Поиск "дружественных имён" СОМ портов На странице: https:/ / norseev. ru/ 2018/ 01/ 04/ comportlist_windows/ нашёл схожую тему. Там приведён код на С++, который показывает только имена СОМ портов, типа,. . .	Сколько Государство потратило денег на меня, обеспечивая инсулином. Programma_Boinc 20.11.2025 Сколько Государство потратило денег на меня, обеспечивая инсулином. Вот решила сделать интересный приблизительный подсчет, сколько государство потратило на меня денег на покупку инсулинов. . . .
Ломающие изменения в C#.NStar Alpha Etyuhibosecyu 20.11.2025 Уже можно не только тестировать, но и пользоваться C#. NStar - писать оконные приложения, содержащие надписи, кнопки, текстовые поля и даже изображения, например, моя игра "Три в ряд" написана на этом. . .	Мысли в слух kumehtar 18.11.2025 Кстати, совсем недавно имел разговор на тему медитаций с людьми. И обнаружил, что они вообще не понимают что такое медитация и зачем она нужна. Самые базовые вещи. Для них это - когда просто люди. . .	Создание Single Page Application на фреймах krapotkin 16.11.2025 Статья исключительно для начинающих. Подходы оригинальностью не блещут. В век Веб все очень привыкли к дизайну Single-Page-Application . Быстренько разберем подход "на фреймах". Мы делаем одну. . .	Фото: Daniel Greenwood kumehtar 13.11.2025	Расскажи мне о Мире, бродяга kumehtar 12.11.2025 — Расскажи мне о Мире, бродяга, Ты же видел моря и метели. Как сменялись короны и стяги, Как эпохи стрелою летели. - Этот мир — это крылья и горы, Снег и пламя, любовь и тревоги, И бескрайние. . .