STL入门

Description

有一家旅馆，共有M个房间，编号为1到M。某天，来了N个客人（编号为1到N）住宿，每个人当然是住在这M个房间中的一个。问题：求出每个房间中的客人的编号？

Input

多组数据。每组数据的第一行输入N M  （1 <= N <= 1000000, 1 <= M <= 10000）第二行有N个数，其中第i个数表示第i个客人所住的房间的编号。
1 <= N <= 1000000

Output

输出M行，第k行表示房间k中的客人的编号（从小到大）。如果k房间中没有客人，输出 -1

• Sample Input
• Raw

7 41 2 3 3 2 1 1

• Sample Output
• Raw

room 1: 1 6 7room 2: 2 5room 3: 3 4room 4: -1

解题思路：由于每个房间里面的人数不确定，所以如果开数组room[10000][1000000]的话太浪费空间。所以要用不定长数组vector。注意输入输出就行了。

#include<iostream>#include<vector>using namespace std;const int maxn=100005;vector<int>v[maxn];int main (){    int N,M,room;    while(~scanf("%d%d",&N,&M)){        for(int i=1;i<N+1;i++){            scanf("%d",&room);            v[room].push_back(i);        }        for(int i=1;i<=M;i++){            printf("room %d:",i);            if(v[i].empty())                printf(" -1\n");            else{                for(vector<int>::iterator it=v[i].begin();it!=v[i].end();it++)                    printf(" %d",*it);                    printf("\n");            }        }        for(int i=0;i<=M;i++)            v[i].clear();    }    return 0;}