力扣刷题记录17.1-----454. 四数相加 II-CSDN博客

本文链接：https://blog.csdn.net/taiyuezyh/article/details/126550500

一、题目

在这里插入图片描述

二、代码

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {


    int nums_size=nums1.size();
    
    std::cout<<"nums_size  "<<nums_size<<std::endl;
    int i,j;
    int temp_sum=0;

    int now_num=0;
    int target_num=0;

    int return_count=0;

    map<int,int> sum_of_12; //默认构造
    map<int,int> sum_of_34; //默认构造
    map<int, int>::iterator it;// 当前map的迭代器

    map<int, int>::iterator target_it;// 目标map的迭代器


    
    for(i=0;i<nums_size;i++)
    {
      for(j=0;j<nums_size;j++)
     {
        temp_sum=nums1[i]+nums2[j];  //计算两项之和
        it=sum_of_12.find(temp_sum); 
        //如果找到 该和加一
        if(it!=sum_of_12.end())
        {
           it->second=it->second+1;
        }
        //如果没找到  在尾部插入
        else
        {
           sum_of_12.insert(pair<int, int>(temp_sum, 1));
        }
     }
    }
    
    // for (map<int, int>::iterator show_it = sum_of_12.begin(); show_it != sum_of_12.end(); show_it++)
	// {
	// 	cout << "key = " << show_it->first << " value = " << show_it->second << endl;
	// }

    for(i=0;i<nums_size;i++)
    {
      for(j=0;j<nums_size;j++)
     {
        temp_sum=nums3[i]+nums4[j];  //计算两项之和
        it=sum_of_34.find(temp_sum); 
        //如果找到 该和加一
        if(it!=sum_of_34.end())
        {
           it->second=it->second+1;
        }
        //如果没找到  在尾部插入
        else
        {
           sum_of_34.insert(pair<int, int>(temp_sum, 1));
        }
     }
    }

    //  for (map<int, int>::iterator show_it = sum_of_34.begin(); show_it != sum_of_34.end(); show_it++)
	// {
	// 	cout << "key = " << show_it->first << " value = " << show_it->second << endl;
	// }
       
       
	for (map<int, int>::iterator it = sum_of_12.begin(); it != sum_of_12.end(); it++)
	{
		//cout << "key = " << it->first << " value = " << it->second << endl;
        now_num=it->first;
        target_num=0-now_num;

        target_it=sum_of_34.find(target_num);  //寻找是否存在

        if(target_it!=sum_of_34.end())   //如果找到目标值
        {
          return_count=return_count+target_it->second*it->second;   //次数 +=12*34

        }
	}



    return return_count;


    }
};