พิจารณาว่าเรามีองค์ประกอบบางอย่างในรายการที่เชื่อมโยง เราต้องหาผลคูณของจำนวน n ตัวสุดท้าย ค่าของ n ยังได้รับ ดังนั้นหากรายการเป็นแบบ [5, 7, 3, 5, 6, 9] และ n =3 ผลลัพธ์จะเป็น 5 * 6 * 9 =270
กระบวนการนี้ตรงไปตรงมา เราเพียงแค่อ่านองค์ประกอบปัจจุบันโดยเริ่มจากด้านซ้าย จากนั้นเพิ่มองค์ประกอบลงในสแต็ก หลังจากเติมสแต็กแล้ว ให้นำองค์ประกอบ n ออกแล้วคูณด้วย prod (เริ่มต้นคือ 1) เมื่อผ่านองค์ประกอบ n จำนวน n ให้หยุด
ตัวอย่าง
#include<iostream>
#include<stack>
using namespace std;
class Node{
public:
int data;
Node *next;
};
Node* getNode(int data){
Node *newNode = new Node;
newNode->data = data;
newNode->next = NULL;
return newNode;
}
void append(struct Node** start, int key) {
Node* new_node = getNode(key);
Node *p = (*start);
if(p == NULL){
(*start) = new_node;
return;
}
while(p->next != NULL){
p = p->next;
}
p->next = new_node;
}
long long prodLastNElements(Node *start, int n) {
if(n <= 0)
return 0;
stack<int> stk;
long long res = 1;
Node* temp = start;
while (temp != NULL) {
stk.push(temp->data);
temp = temp->next;
}
while(n--){
res *= stk.top();
stk.pop();
}
return res;
}
int main() {
Node *start = NULL;
int arr[] = {5, 7, 3, 5, 6, 9};
int size = sizeof(arr)/sizeof(arr[0]);
int n = 3;
for(int i = 0; i<size; i++){
append(&start, arr[i]);
}
cout << "Product of last n elements: " << prodLastNElements(start, n);
} ผลลัพธ์
Product of last n elements: 270
