OJ_1_删除一个链表的倒数第k个结点

#include <iostream>
#include <string>
#include <cmath>
#include <vector>
using namespace std;



struct ListNode{
    int val;
    ListNode* next;
};

int main()
{

    int node_num;
    while(cin>>node_num){
        ListNode * temp,*query;
        if(node_num == 0)   //如果输入数量为0，直接跳过该轮循环
            continue;

        //单独处理头结点
        int head_node_val;
        cin>>head_node_val;
        ListNode *head = new ListNode;
        head->val = head_node_val;
        head->next = nullptr;
        query = head;
        //继续构造链表
        for(int i= 0; i<node_num-1;i++){
            int input;
            cin>>input;
            temp = new ListNode;
            temp->val = input;
            temp->next = nullptr;

            query->next = temp;
            query = temp;

        }


        //倒数第k个结点（1为基），所以是正向遍历的第node_num-k+1个结点（1为基）
        int k;
        cin>> k;
        if(k==0){         //非法输入
            cout<<"0"<<endl;
            continue;
        }

        if(node_num<k)    //k太大了
        {
            cout<<"NULL"<<endl;
            continue;
        }

        query = head;  //遍历用的指针，初始值为头结点
        if(node_num-k == 0){      //如果正好是第一个结点，直接返回
            cout<<head->val<<endl;
            continue;
        }


        for(int i=1; i<(node_num-k+1);i++){
            query = query->next;
        }
        cout<<query->val<<endl;
    }
    return 0;
}