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

@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];
}

Новые блоги и статьи Все статьи Все блоги /
SDL3 для Web (WebAssembly): Реализация движения на Box2D v3 - трение и коллизии с повёрнутыми стенами 8Observer8 20.02.2026 Содержание блога Box2D позволяет легко создать главного героя, который не проходит сквозь стены и перемещается с заданным трением о препятствия, которые можно располагать под углом, как верхнее. . .	Конвертировать закладки radiotray-ng в m3u-плейлист damix 19.02.2026 Это можно сделать скриптом для PowerShell. Использование . \СonvertRadiotrayToM3U. ps1 <path_to_bookmarks. json> Рядом с файлом bookmarks. json появится файл bookmarks. m3u с результатом. # Check if. . .	Семь CDC на одном интерфейсе: 5 U[S]ARTов, 1 CAN и 1 SSI Eddy_Em 18.02.2026 Постепенно допиливаю свою "многоинтерфейсную плату". Выглядит вот так: https:/ / www. cyberforum. ru/ blog_attachment. php?attachmentid=11617&stc=1&d=1771445347 Основана на STM32F303RBT6. На борту пять. . .	Камера Toupcam IUA500KMA Eddy_Em 12.02.2026 Т. к. у всяких "хикроботов" слишком уж мелкий пиксель, для подсмотра в ESPriF они вообще плохо годятся: уже 14 величину можно рассмотреть еле-еле лишь на экспозициях под 3 секунды (а то и больше),. . .
И ясному Солнцу zbw 12.02.2026 И ясному Солнцу, и светлой Луне. В мире покоя нет и люди не могут жить в тишине. А жить им немного лет.	«Знание-Сила» zbw 12.02.2026 «Знание-Сила» «Время-Деньги» «Деньги -Пуля»	SDL3 для Web (WebAssembly): Подключение Box2D v3, физика и отрисовка коллайдеров 8Observer8 12.02.2026 Содержание блога Box2D - это библиотека для 2D физики для анимаций и игр. С её помощью можно определять были ли коллизии между конкретными объектами и вызывать обработчики событий столкновения. . . .	SDL3 для Web (WebAssembly): Загрузка PNG с прозрачным фоном с помощью SDL_LoadPNG (без SDL3_image) 8Observer8 11.02.2026 Содержание блога Библиотека SDL3 содержит встроенные инструменты для базовой работы с изображениями - без использования библиотеки SDL3_image. Пошагово создадим проект для загрузки изображения. . .