結果

問題 No.742 にゃんにゃんにゃん　猫の挨拶
コンテスト
ユーザー MSK
提出日時 2018-10-10 13:56:18
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
TLE  
実行時間 -
コード長 1,224 bytes
コンパイル時間 573 ms
コンパイル使用メモリ 63,208 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-10-12 17:15:16
合計ジャッジ時間 5,405 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 11 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(void) {
  int N;
  cin >> N;
  vector<int> M(N + 1);
  for(int i = 1; i <= N; i++) {
    cin >> M[i];
  }
  vector<int> V(N + 1, 0);
  for(int i = 1; i <= N; i++) {
    if(0 < M[i] - i) {
      V[i] = 1;
    } else if(0 > M[i] - i) {
      V[i] = -1;
    }
  }
  int maxt = 0;
  for(int i = 1; i <= N; i++) {
    maxt = max(maxt, abs(M[i] - i));
  }
  int ans = 0;
  for(int t = 1; t <= maxt; t++) {
    vector<int> NCatStay(30001, 0);
    vector<int> NCatComeFromL(30001, 0);
    vector<int> NCatComeFromR(30001, 0);
    for(int i = 1; i <= N; i++) {
      if(0 == V[i]) {
        NCatStay[M[i]]++;
      } else {
        int pos = i + V[i] * t;
        if(V[i] > 0) {
          NCatComeFromL[pos]++;
        } else {
          NCatComeFromR[pos]++;
        }
        if(M[i] == pos) {
          V[i] = 0;
        }
      }
    }
    for(int i = 1; i < 30001; i++) {
      int NCatCome = NCatComeFromL[i] + NCatComeFromR[i];
      ans += NCatStay[i] * NCatCome;
      ans += NCatCome * (NCatCome - 1) / 2;
      if(i > 1) {
        ans += NCatComeFromR[i - 1] * NCatComeFromL[i];
      }
    }
  }
  cout << ans << endl;
  return 0;
}
0