c++糖果

MIS R有 m 颗糖，编号为 1,2,⋯ ,m。他打算发给 n个小朋友。发完糖之后，为了促进小朋友之间的友谊，他会安排 r 次互换活动。每次互换活动，会让第 x 个小朋友和第 y个小朋友交换手里的糖果。现在SSR想知道最终每个小朋友手中有哪些糖果。

输入格式

第一行两个整数 n,m,,r（2≤n≤1000，1≤m,r≤10^6 ），表示 n个小朋友，m 颗糖，r 次互换安排。

第二行 有m 个整数，表示第i个糖果发给了第ai 个小朋友。

接下来 r行，每行两个整数 x,y，表示第 x个小朋友和第 y个小朋友互换手中的糖果。

数据保证 1≤ai,x,y≤n，x≠y。

输出格式

n 行，每行以一个整数 ci开头，表示第 i个小朋友含有的糖果数，接下来是一个长度为 ci的递增编号序列，整数之间用空格隔开，表示第 i个小朋友手中糖果的编号。

输入/输出例子1

输入：

3 6 1

1 3 2 1 3 2

2 3

输出：

2 1 4

2 2 5

2 3 6

#include<bits/stdc++.h>
using namespace std;
vector<int>a[10001];
int main(){
    int n,m,r;
    cin>>m>>n>>r;
    for(int i=1;i<=n;i++)
    {
        int t;
        cin>>t;
        a[t].push_back(i);
    }
    for(int i=1;i<=r;i++)
    {
        int x,y;
        cin>>x>>y;
        a[x].swap(a[y]);
    }
    for(int i=1;i<=m;i++)
    {
        cout<<a[i].size()<<" ";
        for(int j=0;j<a[i].size();j++)
        cout<<a[i][j]<<" ";
        cout<<"\n";
    }
    return 0;
}