bzoj1483: [HNOI2009]夢幻布丁(vector+啟發式合併)
阿新 • • 發佈：2018-11-02
1483: [HNOI2009]夢幻布丁
Time Limit: 10 Sec Memory Limit: 64 MBSubmit: 4022 Solved: 1640
[Submit][Status][Discuss]
Description
N個布丁擺成一行,進行M次操作.每次將某個顏色的布丁全部變成另一種顏色的,然後再詢問當前一共有多少段顏色. 例如顏色分別為1,2,2,1的四個布丁一共有3段顏色.Input
第一行給出N,M表示布丁的個數和好友的操作次數.Output
針對第二類操作即詢問，依次輸出當前有多少段顏色.
Sample Input
4 31 2 2 1
2
1 2 1
2
Sample Output
31
/*vector 記錄每個顏色出現的位置 修改時啟發式合併，更細答案即可。 */ #include<bits/stdc++.h> #define N 1000007 #define inf 0x3f3f3f3f #define ll long long using namespace std; vector<int>v[N]; vector<int>::iterator it; int n,m; int fa[N],a[N]; int main() { scanf("%d%d",&n,&m); intans=0; for(int i=1;i<=n;i++) { scanf("%d",&a[i]); if(a[i]!=a[i-1])ans++; fa[a[i]]=a[i]; v[a[i]].push_back(i); } while(m--) { int t; scanf("%d",&t); if(t==2) printf("%d\n",ans); else { int x,z; scanf("%d%d",&x,&z); if(x==z)continue; if(v[fa[x]].size()>v[fa[z]].size()) swap(fa[x],fa[z]); int col1=fa[x],col2=fa[z]; for(int i=0;i<v[col1].size();i++) { v[col2].push_back(v[col1][i]); if(a[v[col1][i]-1]==col2) ans--; if(a[v[col1][i]+1]==col2) ans--; } for(int i=0;i!=v[col1].size();i++) a[v[col1][i]]=col2; v[col1].clear(); } } return 0; }