fork download
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define rep(i,n) for(int(i)=0 ; (i)<(int)(n) ; ++(i))
  4. #define replu(i,l,u) for(int(i)=(int)(l) ; (i)<=(int)(u) ; ++(i))
  5. #define repul(i,u,l) for(int(i)=(int)(u) ; (i)>=(int)(l) ; --(i))
  6. #define fi first
  7. #define se second
  8. #define pb push_back
  9. typedef vector<int> vi ;
  10. typedef vector<vector<int> > vvi ;
  11. typedef pair<int,int> pii ;
  12. typedef pair<int, pii> pipii ;
  13. typedef vector<pair<int,int> > vpii ;
  14. typedef long long ll ;
  15. const int mod=1e9+7 ;
  16. const ll inf=2e9+10 ;
  17. template <typename T1, typename T2> static void relax(T1 &x, T2 y) { if(x>y) x = y ; }
  18. template <typename T1, typename T2> static void gain(T1 &x, T2 y) { if(x<y) x = y ; }
  19. ll powmod(ll a, ll b) { ll res = 1 ; while(b) { if(b&1) res = (res*a)%mod ; a = (a*a)%mod ; b >>= 1 ; } return res ; }
  20. ll gcd(ll a, ll b) { return b ? gcd(b, a%b) : a ; }
  21.  
  22. const int maxn = 3e5 + 5 ;
  23. int n ;
  24. string a[maxn] ;
  25. pii f[maxn] ;
  26. int main() {
  27. ios::sync_with_stdio(false) ;
  28. cin.tie(NULL) ;
  29. cin >> n ;
  30. rep(i,n) cin >> a[i] ;
  31. int op, cl ;
  32. rep(i,n) {
  33. op = cl = 0 ;
  34. int l = (int)a[i].length() ;
  35. rep(j,l) {
  36. if(a[i][j] == '(') op++ ;
  37. else { if(op > 0) op-- ; else cl++ ; }
  38. }
  39. f[i] = {op, cl} ;
  40. }
  41. int cntop[maxn], cntcl[op] ;
  42. rep(i,maxn) cntop[i] = cntcl[i] = 0 ;
  43. ll ans = 0 ;
  44. rep(i,n) if(f[i].fi == 0 && f[i].se == 0) ans++ ;
  45. ans = ans*ans ;
  46. rep(i,n) {
  47. if(f[i].se == 0 && f[i].fi != 0) {
  48. ans += cntcl[f[i].fi]*1LL ; cntop[f[i].fi]++ ;
  49. }
  50. if(f[i].fi == 0 && f[i].se != 0) {
  51. ans += cntop[f[i].se]*1LL ; cntcl[f[i].se]++ ;
  52. }
  53. }
  54. cout << ans << "\n" ;
  55. return 0 ;
  56. }
Success #stdin #stdout 0.01s 13640KB
stdin
3
)
()
(
stdout
2