查找函数的调用关系

// mytest.cpp : 定义控制台应用程序的入口点。

//

#include "stdafx.h"

#include <stdio.h>

#include <sys/types.h>

#include <sys/stat.h>   

#include <fcntl.h>

#include <iostream>

#include <fstream>

#include <string>

#include <vector>

#include <algorithm>

#include <list>

#include <map>

using namespace std;

#define MAINFUNID 0

typedef map<int, vector<int>> MAP_INT_VEC;

typedef list<vector<int>> LIST_VEC;

void PrintVector(vector<int> &vecFuncNodeId)

{

    vector<int>::iterator it;

    for (it = vecFuncNodeId.begin(); it != vecFuncNodeId.end(); it++ )

    {

        cout<<*it<<" ";

    }

    cout<<endl<<endl;

}

//解析函数ID

void GetFuncNodeIdByLine(string strline, vector<int> &vecFuncNodeId)

{

    int uiFuncNodeId = 0;

    string strfunid;

    while( strline.find(' ') != string::npos)

    {

        strfunid = strline.substr(0, strline.find(' '));

        uiFuncNodeId = atoi(strfunid.c_str());

        #if 0

        //cout<<"int fun id: " << uiFuncNodeId<<endl;

        if (uiFuncNodeId == 0)

        {

            vecFuncNodeId.insert(vecFuncNodeId.begin(),uiFuncNodeId);

        }

        else

        {

            vecFuncNodeId.push_back(uiFuncNodeId);

        }

        #endif

       

        vecFuncNodeId.push_back(uiFuncNodeId);

       

        strline = strline.substr(strline.find(' ')+1);

        //cout<<"strline:"<<strline<<endl; 

    }

    //cout<<"strline: "<<strline<<endl;

    //题目保证了每行末尾没有空格

    uiFuncNodeId = atoi(strline.c_str());

    //cout<<"int fun id: " << uiFuncNodeId<<endl;

    vecFuncNodeId.push_back(uiFuncNodeId);

}

//被调函数是主函数，删除该元素

void DeleteElemOfVec(vector<int> &vecFunId, int elem)

{

    vector<int>::iterator iter = vecFunId.begin();

    for (iter = vecFunId.begin(); iter != vecFunId.end();)

    {

        if(*iter == elem)

        {

            iter  = vecFunId.erase(iter);

        }

        else

        {

            iter++;

        }

        if(iter == vecFunId.end()) //要控制迭代器不能超过整个容器

        { 

            break;

        } 

    }  

}

//去除相邻的重复元素去除相邻的重复元素

void DistinctVector(vector<int> &vec)

{

    vector<int>::iterator itNewEnd;

    itNewEnd = unique(vec.begin(), vec.end());

    vec.erase(itNewEnd, vec.end());

}

//合并两个vector容器

void CombineVector(vector<int> &vec1, vector<int> &vec2, vector<int> &vec3)

{  

    if (!vec1.empty())

    {

        //从vec3的首地址插入vec1 首尾间元素

        vec3.insert(vec3.begin(), vec1.begin(), vec1.end());

    }

    if (!vec2.empty())

    {

        //从vec3的尾地址插入vec2首尾间元素

        vec3.insert(vec3.end(), vec2.begin(), vec2.end());

    }

    //vec3默认升序排序

    sort(vec3.begin(), vec3.end());

    //去掉vec3中的相邻的重复元素

    DistinctVector(vec3);

}

//获取该函数的调用关系

void GetFuncCallRelation(string strline, MAP_INT_VEC &mapFunCallRelate)

{

    int uiFuncCallId = 0;

    vector<int> vecFuncNodeId;

    vector<int> vecFuncBeCalledNodeId;

    GetFuncNodeIdByLine(strline, vecFuncNodeId);

    PrintVector(vecFuncNodeId);

    if (!vecFuncNodeId.empty())

    {

        //获取调用函数

        uiFuncCallId = vecFuncNodeId[0];

        #if 0

        //去除相邻的重复元素去除相邻的重复元素

         vector<int>::iterator itNewEnd;

         itNewEnd = unique(vecFuncNodeId.begin(), vecFuncNodeId.end());

         vecFuncNodeId.erase(itNewEnd, vecFuncNodeId.end());

         PrintVector(vecFuncNodeId);

         #endif

        

         for (vector<int>::iterator it= vecFuncNodeId.begin() + 1; it != vecFuncNodeId.end(); it++)

         {

            //cout << *it<<endl;

            vecFuncBeCalledNodeId.push_back(*it);      

         }

         //vector 默认是按升序排列,即从小到大

         sort(vecFuncBeCalledNodeId.begin(), vecFuncBeCalledNodeId.end());

         //删除被调函数容器中的主函数ID

         DeleteElemOfVec(vecFuncBeCalledNodeId, MAINFUNID);

         //去除相邻的重复元素去除相邻的重复元素

         DistinctVector(vecFuncBeCalledNodeId);

         PrintVector(vecFuncBeCalledNodeId);

    }

    //判断调用函数在map容器中是否存在，如存在，合并被调函数

    MAP_INT_VEC::iterator it = mapFunCallRelate.find(uiFuncCallId);

    if (it == mapFunCallRelate.end())

    {

        mapFunCallRelate[uiFuncCallId] = vecFuncBeCalledNodeId;

    }

    else

    {

        vector<int> vecCombineElem;

        CombineVector(mapFunCallRelate[uiFuncCallId] ,vecFuncBeCalledNodeId, vecCombineElem);

        mapFunCallRelate[uiFuncCallId] = vecCombineElem;

    }

   

    //PrintVector(vecFuncCallRelation);

}

//从txt 文件中读取数据

void ReadDataByFileToFunCallRelate(string tmpstr, MAP_INT_VEC &mapFunCallRelate)

{

    int line = 1;

    vector<int> vecFuctId;

    //vector<int> vecFuctCallRelation;

   

    //char str[100];

    ifstream infile("D://VisualStudio2010/Projects/mytest/mytest/input.txt");

    cout<<"Read from file:" << endl;

    if (! infile.is_open())

    {

        cout<<"Error opening file";

        exit(1);

    }

    while(!infile.eof())

    {

        getline(infile, tmpstr);

        //cout <<line<<": "<<tmpstr.c_str()<<endl;

        line++;

        //GetFuncNodeIdByLine(tmpstr, vecFuctId);

        GetFuncCallRelation(tmpstr,mapFunCallRelate);

    }

    infile.close();

    //PrintVector(vecFuctId);

}

int  FindFunCallRelation(MAP_INT_VEC &mapFunCallRelate, vector<int> &vecFunRelate)

{

    //LIST_VEC ListBeCalledFunVec;

    int FunCallNodeId = 0;

    int FunNodeId = 0;

    list<int> listFunNodeId;

    vector<int> vecBeCalledId;

    if (!mapFunCallRelate.empty())

    {

        //将map容器的的key存在list容器中

        MAP_INT_VEC::iterator iter;

        for (iter = mapFunCallRelate.begin(); iter != mapFunCallRelate.end(); iter++)

        {

            FunCallNodeId = iter->first;

            cout << "callnodeid: "<<FunCallNodeId<<endl;

            vecBeCalledId = iter->second;

            for (int i = 0; i < vecBeCalledId.size(); i++)

            {

                cout << vecBeCalledId[i] << " ";

            }

            cout<<endl;

       

            listFunNodeId.push_back(FunCallNodeId);

        }

       

        FunNodeId = listFunNodeId.front();

        vecFunRelate.push_back(FunNodeId);

        //map key默认从小到大排序,第一个元素的key不为主函数，则该关系中午主函数

        if (FunNodeId != MAINFUNID)

        {          

            return -1;

        }

       

        cout<<endl<<"FunNodeId: ";

        for (list<int>::iterator iter = listFunNodeId.begin(); iter != listFunNodeId.end(); )

        {  

            FunNodeId = listFunNodeId.front();

            listFunNodeId.remove(*iter++);

            //cout<<FunNodeId<<" ";

#if 1

            if (mapFunCallRelate.find(FunNodeId) != mapFunCallRelate.end())

            {              

                for(int i = 0 ; i < mapFunCallRelate[FunNodeId].size(); i++)

                {

                    if (find(listFunNodeId.begin(), listFunNodeId.end(), mapFunCallRelate[FunNodeId][i]) != listFunNodeId.end())

                    {

                        cout << mapFunCallRelate[FunNodeId][i] << " ";

                       

                        vecFunRelate.push_back(
ac56
mapFunCallRelate[FunNodeId][i]);

                    }

                    else

                    {

                        continue;

                    }

                }

            }

            else

            {

                continue;

            }

#endif

        }

        cout<<endl;

    }

   

    return 0;

}

char* szOutputFileName = "D://VisualStudio2010/Projects/mytest/mytest/output.txt";

void WriteDataToFile(vector<int> &vecData)

{

    ofstream outputfile(szOutputFileName);

    if (outputfile.is_open())

    {

        cout<<"Function call relation:  "<<endl;

        for (int i = 0 ; i < vecData.size(); i++)

        {

            outputfile<<vecData[i]<<endl;

        }

        outputfile.close();

    }

}

int _tmain(int argc, _TCHAR* argv[])

{

    string linestr;

    vector<int> vecFunRelate;

    MAP_INT_VEC mapFunCallRelate;

    ReadDataByFileToFunCallRelate(linestr, mapFunCallRelate);  

    FindFunCallRelation(mapFunCallRelate, vecFunRelate);

    PrintVector(vecFunRelate);

    WriteDataToFile(vecFunRelate);

    return 0;

}

 

 

int  FindFunCallRelation(MAP_INT_VEC &mapFunCallRelate, vector<vector<int>> &vecFunRelate)

{

 int iFunCallNodeId = 0;

 int FunNodeId = 0;

 int index = 0;

 list<int> listCallFunNodeId;

 vector<int> vecBeCalledId;

 vector<int> vecMainCallingId;

 

 if (!mapFunCallRelate.empty())

 {

  //将map容器的的key存在list容器中

  //map的首key存于vertor容器,key默认从小到大排序,第一个元素的key不为主函数，则该关系中午主函数

  MAP_INT_VEC::iterator itLocate = mapFunCallRelate.find(MAINFUNID);

  if (itLocate != mapFunCallRelate.end())

  {

   vecMainCallingId.push_back(MAINFUNID);

   vecFunRelate.push_back(vecMainCallingId);

   listCallFunNodeId.push_back(MAINFUNID);   

  }

  else

  {

   return -1;  

  }

  cout << "--------------------------------------------------------------------------"<<endl;

  MAP_INT_VEC::iterator iter;

  for (iter = mapFunCallRelate.begin(); iter != mapFunCallRelate.end(); iter++)

  {

   

   iFunCallNodeId = iter->first;

   cout << "callnodeid: "<<iFunCallNodeId<<endl;

   vecBeCalledId = iter->second;

   for (int i = 0; i < vecBeCalledId.size(); i++)

   {

    cout << vecBeCalledId[i] << " ";

   }

   cout<<endl;

  

   //listCallFunNodeId.push_back(iFunCallNodeId);

  }

  cout << "--------------------------------------------------------------------------"<<endl; 

  

  //MAP_INT_VEC::iterator iter;

  for (iter = mapFunCallRelate.begin(); iter != mapFunCallRelate.end(); iter++)

  {

   vector<int> vecCallingId;

   iFunCallNodeId = iter->first;

   cout <<"first value - "<<iFunCallNodeId <<" ;";

   vecBeCalledId = iter->second;

   for (int i = 0; i < vecBeCalledId.size(); i++)

   { 

    //cout << i << " ";  

    if (mapFunCallRelate.find(vecBeCalledId[i]) != mapFunCallRelate.end() && find(listCallFunNodeId.begin(), listCallFunNodeId.end(), vecBeCalledId[i]) == listCallFunNodeId.end())

    {

     cout << vecBeCalledId[i] << " ";

     vecCallingId.push_back(vecBeCalledId[i]);

     listCallFunNodeId.push_back(vecBeCalledId[i]);

    }

    else

    {

     continue;

    }

    

   }

   vecFunRelate.push_back(vecCallingId);

   cout<<endl;

  }

  cout << "--------------------------------------------------------------------------"<<endl; 

  cout<<"output result: "<<endl;

  for (list<int>::iterator iter = listCallFunNodeId.begin(); iter != listCallFunNodeId.end(); iter++)

  {

   cout<<*iter<<" ";

  }

  cout<<endl;

 }

 else

 {

  return -1;

 }

 

 return 0;

}

char* szOutputFileName = "D://VisualStudio2010/Projects/mytest/mytest/output.txt";

void WriteDataToFile(vector<vector<int>> &vecData)

{

 ofstream outputfile(szOutputFileName);

 if (outputfile.is_open())

 {

  cout<<"Function call relation:  "<<endl;

  for (int i = 0 ; i < vecData.size(); i++)

  {

   for (int j = 0; j < vecData[i].size(); j++)

   {

    outputfile<<vecData[i][j]<<" ";

    cout<<vecData[i][j]<<" ";

   }

   cout<<endl;

   outputfile<<endl;

  }

  outputfile.close();

 }

}

int _tmain(int argc, _TCHAR* argv[])

{

 int ret;

 string linestr;

 vector<int> vecFunRelate;

 vector<vector<int>> vecOutputFunRelate;

 MAP_INT_VEC mapFunCallRelate;

 ReadDataByFileToFunCallRelate(linestr, mapFunCallRelate); 

 ret = FindFunCallRelation(mapFunCallRelate, vecOutputFunRelate);

 if (ret == 0)

 {

  WriteDataToFile(vecOutputFunRelate);   

 }

 

 PrintVector(vecFunRelate);

   

 return 0;

}