【清橙 A1206】小Z的袜子(莫队算法)
代码如下:
#include <iostream> #include <cmath> #include <vector> #include <cstdlib> #include <cstdio> #include <cstring> #include <queue> #include <list> #include <algorithm> #include <map> #include <set> #define LL long long #define Pr pair<int,int> #define fread() freopen("in.in","r",stdin) #define fwrite() freopen("out.out","w",stdout) using namespace std; const int INF = 0x3f3f3f3f; const int msz = 10000; const double eps = 1e-8; //分块 int pos[55555]; struct Range { LL l,r,id; bool operator < (const struct Range a) const { return pos[l] == pos[a.l]? r < a.r: pos[l] < pos[a.l]; } }; Range rg[55555]; LL wz[55555]; LL cnt[55555]; LL up[55555],down[55555]; LL ans; void update(int x,int d) { ans -= cnt[wz[x]]*cnt[wz[x]]; cnt[wz[x]] += d; ans += cnt[wz[x]]*cnt[wz[x]]; } int main() { int n,m; while(~scanf("%d%d",&n,&m)) { int bk = ceil(sqrt(n*1.0)); for(int i = 1; i <= n; ++i) pos[i] = (i-1)/bk; for(int i = 1; i <= n; ++i) scanf("%lld",&wz[i]); for(int i = 0; i < m; ++i) { scanf("%lld%lld",&rg[i].l,&rg[i].r); rg[i].id = i; } //分块排序 sort(rg,rg+m); memset(cnt,0,sizeof(cnt)); LL l = 1, r = 0; ans = 0; for(int i = 0; i < m; ++i) { int id = rg[i].id; if(rg[i].l == rg[i].r) { up[id] = 0; down[id] = 1; continue; } while(r < rg[i].r) update(++r,1); while(r > rg[i].r) update(r--,-1); while(l > rg[i].l) update(--l,1); while(l < rg[i].l) update(l++,-1); up[id] = ans-r+l-1; down[id] = (r-l+1)*(r-l); LL c = __gcd(up[id],down[id]); up[id] /= c; down[id] /= c; } for(int i = 0; i < m; ++i) printf("%lld/%lld\n",up[i],down[i]); } return 0; }