正題

nowcoder 1103A

---

題目大意

有一個數組，將其復制k遍，求所有區間內不同元素個數的和

---

解題思路

同一個區間內有相同的數那么在最右邊的數計算貢獻

記 $r$ 為右邊第一個相等的數，那么貢獻就是到 $r$ 的距離乘上左邊的數字個數（有哪些區間可以讓這個數計算貢獻）

---

code

#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define ll long long #define wyc 1000000007 #define N 100100 using namespace std; ll n,k,m,gg,ans,a[N],b[N],L[N],rs[N]; ll ksm(ll x,ll y) {ll z=1;while(y){if(y&1)z=z*x%wyc;x=x*x%wyc;y>>=1;}return z; } int main() {scanf("%lld%lld",&n,&k);for(ll i=1;i<=n;++i){scanf("%lld",&a[i]);b[i]=a[i];}sort(b+1,b+1+n);m=unique(b+1,b+1+n)-b-1;for(ll i=1;i<=n;++i)a[i]=lower_bound(b+1,b+1+m,a[i])-b;for(ll i=1;i<=m;++i)L[i]=n+1;for(ll i=n;i>0;--i){rs[i]=L[a[i]];L[a[i]]=i;}gg=ksm(2,wyc-2);for(ll i=1;i<=n;++i)ans=(ans+(i+n*(k-1)%wyc)%wyc*(rs[i]-i)%wyc)%wyc;for(ll i=1;i<=n;++i){if(rs[i]==n+1)rs[i]=n+L[a[i]];ans=(ans+(i+i+(k-2)*n%wyc)%wyc*(k-1)%wyc*gg%wyc*(rs[i]-i)%wyc)%wyc;}printf("%lld",ans);return 0; } 創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的复读数组（nowcoder 1103A）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。